diff --git a/aimodel/src/lib/dataset/dataset_segmenter.py b/aimodel/src/lib/dataset/dataset_segmenter.py
index d191bca..2247c4f 100644
--- a/aimodel/src/lib/dataset/dataset_segmenter.py
+++ b/aimodel/src/lib/dataset/dataset_segmenter.py
@@ -3,16 +3,13 @@ import math
 import json
 
 from loguru import logger
-
 import tensorflow as tf
 
 from lib.dataset.read_metadata import read_metadata
-
 from ..io.readfile import readfile
 from .shuffle import shuffle
 
 
-
 # TO PARSE:
 def parse_item(metadata, shape_water_desired, water_threshold=0.1):
 	water_width_source, water_height_source, _water_channels_source = metadata["waterdepth"]