black linting for cli resumable training

talmolab · Jan 19, 2023 · 9b5f163 · 9b5f163
1 parent 34dbbdb
commit 9b5f163
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 14 deletions.
diff --git a/sleap/nn/training.py b/sleap/nn/training.py
@@ -1871,16 +1871,12 @@ def main(args: Optional[List] = None):
     parser.add_argument(
         "--resume",
         action="store_true",
-        help=(
-            "Resume training from last checkpoint."
-        ),
+        help=("Resume training from last checkpoint."),
     )
     parser.add_argument(
         "--base_checkpoint",
         default="",
-        help=(
-            "Path to base checkpoint to resume training from."
-        ),
+        help=("Path to base checkpoint to resume training from."),
     )
 
     device_group = parser.add_mutually_exclusive_group(required=False)

diff --git a/tests/nn/test_training.py b/tests/nn/test_training.py
@@ -310,6 +310,7 @@ def test_train_cropping(
         == 0
     )
 
+
 def test_resume_training_cli(tmp_path, training_labels, cfg):
     """
     Test CLI to resume training.
@@ -319,14 +320,24 @@ def test_resume_training_cli(tmp_path, training_labels, cfg):
 
     # path to sleap/tests/data/models/minimal_robot.UNet.single_instance
     base_checkpoint_path = os.path.join(
-        os.path.dirname(__file__), "..", "data", "models", "minimal_robot.UNet.single_instance"
-    )
-    json_path = os.path.join(
-        base_checkpoint_path, "training_config.json"
-    )
-    labels_path = os.path.join(
-        base_checkpoint_path, "labels_gt.slp"
+        os.path.dirname(__file__),
+        "..",
+        "data",
+        "models",
+        "minimal_robot.UNet.single_instance",
     )
+    json_path = os.path.join(base_checkpoint_path, "training_config.json")
+    labels_path = os.path.join(base_checkpoint_path, "labels_gt.slp")
 
     # run CLI to resume training
-    main(["--training_job_path", json_path, "--labels_path", labels_path, "--resume", "--base_checkpoint", base_checkpoint_path])
+    main(
+        [
+            "--training_job_path",
+            json_path,
+            "--labels_path",
+            labels_path,
+            "--resume",
+            "--base_checkpoint",
+            base_checkpoint_path,
+        ]
+    )