diff --git a/aimodel/src/subcommands/pretrain.py b/aimodel/src/subcommands/pretrain.py
index 865e8c9..2506b8f 100644
--- a/aimodel/src/subcommands/pretrain.py
+++ b/aimodel/src/subcommands/pretrain.py
@@ -47,7 +47,7 @@ def run(args):
 	# 	print("LABEL", label.shape)
 	# print("ITEMS DONE")
 	# exit(0)
-
+	
 	
 	ai = RainfallWaterContraster(
 		dir_output=args.output,