diff --git a/mu_map/training/default.py b/mu_map/training/default.py
index fd554ffd49b9c25c88b0a70855a38433b4db3048..8e092202cb84d09eece6b18f0f6ae1073d024978 100644
--- a/mu_map/training/default.py
+++ b/mu_map/training/default.py
@@ -162,24 +162,24 @@ if __name__ == "__main__":
         "--patch_size",
         type=int,
         default=32,
-        help="the size of patches extracted for each reconstruction"
+        help="the size of patches extracted for each reconstruction",
     )
     parser.add_argument(
         "--patch_offset",
         type=int,
         default=20,
-        help="offset to ignore the border of the image"
+        help="offset to ignore the border of the image",
     )
     parser.add_argument(
         "--number_of_patches",
         type=int,
         default=100,
-        help="number of patches extracted for each image"
+        help="number of patches extracted for each image",
     )
     parser.add_argument(
         "--no_shuffle",
-        action="store_true" 
-        help="do not shuffle patches in the dataset"
+        action="store_true",
+        help="do not shuffle patches in the dataset",
     )
 
     # Training Args