black

talmolab · Jul 10, 2024 · f896a88 · f896a88
1 parent 9d3334e
commit f896a88
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/sleap_io/io/main.py b/sleap_io/io/main.py
@@ -58,13 +58,12 @@ def load_nwb(filename: str, format: str) -> Labels:
     """
     if format == "nwb":
         return nwb.read_nwb(filename)
-    
+
     elif format == "nwb_training":
         return
 
 
-def save_nwb(labels: Labels, filename: str, format: str,
-             append: bool = True, **kwargs):
+def save_nwb(labels: Labels, filename: str, format: str, append: bool = True, **kwargs):
     """Save a SLEAP dataset to NWB format.
 
     Args:
@@ -81,7 +80,7 @@ def save_nwb(labels: Labels, filename: str, format: str,
             nwb.append_nwb(labels, filename, **kwargs)
         else:
             nwb.write_nwb(labels, filename, **kwargs)
-    
+
     elif format == "nwb_training":
         raise NotImplementedError
 

diff --git a/sleap_io/io/nwb.py b/sleap_io/io/nwb.py
@@ -108,8 +108,9 @@ def labels_to_pose_training(labels: Labels, **kwargs) -> PoseTraining:  # type:
                 unit="NA",
                 format="external",
                 external_file=[training_frame_video.filename],
-                dimension=[training_frame_video.backend.img_shape[0],
-                           training_frame_video.backend.img_shape[1],
+                dimension=[
+                    training_frame_video.backend.img_shape[0],
+                    training_frame_video.backend.img_shape[1],
                 ],
                 starting_frame=[0],
                 rate=30.0,