Lint

talmolab · Apr 14, 2024 · 8d5ceb1 · 8d5ceb1
1 parent 9fe8191
commit 8d5ceb1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/sleap_io/__init__.py b/sleap_io/__init__.py
@@ -25,5 +25,5 @@
     load_jabs,
     save_jabs,
     load_video,
-    load_file
+    load_file,
 )
diff --git a/sleap_io/io/main.py b/sleap_io/io/main.py
@@ -78,7 +78,7 @@ def load_labelstudio(
 
 def save_labelstudio(labels: Labels, filename: str):
     """Save a SLEAP dataset to Label Studio format.
-    
+
     Args:
         labels: A SLEAP `Labels` object (see `load_slp`).
         filename: Path to save labels to ending with `.json`.
@@ -115,7 +115,7 @@ def save_jabs(labels: Labels, pose_version: int, root_folder: Optional[str] = No
 
 def load_video(filename: str, **kwargs) -> Video:
     """Load a video file.
-    
+
     Args:
         filename: Path to a video file.
 
@@ -145,4 +145,4 @@ def load_file(filename: str, **kwargs) -> Union[Labels, Video]:
     elif filename.endswith(".h5"):
         return load_jabs(filename)
     else:
-        return load_video(filename, **kwargs)
+        return load_video(filename, **kwargs)
diff --git a/tests/io/test_main.py b/tests/io/test_main.py
@@ -10,7 +10,7 @@
     load_jabs,
     save_jabs,
     load_video,
-    load_file
+    load_file,
 )