Format files

Author: gitttt-1234

sleap_nn/predict.py

@@ -510,9 +510,15 @@ def run_inference(
             if video_index is not None and len(output.videos) > video_index:
                 video = output.videos[video_index]
                 # Get video filename and sanitize it for use in path
-                video_name = Path(video.filename).stem if isinstance(video.filename, str) else f"video_{video_index}"
+                video_name = (
+                    Path(video.filename).stem
+                    if isinstance(video.filename, str)
+                    else f"video_{video_index}"
+                )
                 # Insert video name before .predictions.slp extension
-                output_path = base_path.parent / f"{base_path.stem}.{video_name}.predictions.slp"
+                output_path = (
+                    base_path.parent / f"{base_path.stem}.{video_name}.predictions.slp"
+                )
             else:
                 output_path = base_path.with_suffix(".predictions.slp")
         output.save(Path(output_path).as_posix(), restore_original_videos=False)

tests/test_predict.py

@@ -1454,7 +1454,9 @@ def test_video_index_output_path(
     # The output file should contain the video name
     output_file = output_files[0]
     video_name = Path(labels.videos[0].filename).stem
-    assert video_name in output_file.stem, f"Video name '{video_name}' not found in output path '{output_file}'"
+    assert (
+        video_name in output_file.stem
+    ), f"Video name '{video_name}' not found in output path '{output_file}'"
 
     # Clean up
     for f in output_files: