Return video object for debugging

LorenFrankLab · Oct 28, 2024 · 19ff538 · 19ff538
1 parent 65296cf
commit 19ff538
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/spyglass/position/v1/dlc_utils_makevid.py b/src/spyglass/position/v1/dlc_utils_makevid.py
@@ -530,4 +530,4 @@ def concat_partial_videos(self):
 
 def make_video(**kwargs):
     """Passthrough for VideoMaker class for backwards compatibility."""
-    VideoMaker(**kwargs)
+    return VideoMaker(**kwargs)
diff --git a/src/spyglass/position/v1/position_dlc_selection.py b/src/spyglass/position/v1/position_dlc_selection.py
@@ -419,7 +419,7 @@ def make(self, key):
             video_frame_inds = video_frame_inds[:limit]
             pos_info_df = pos_info_df.head(limit)
 
-        make_video(
+        video_maker = make_video(
             video_filename=video_filename,
             video_frame_inds=video_frame_inds,
             position_mean={
@@ -441,6 +441,7 @@ def make(self, key):
         )
 
         if limit:  # don't insert if we're just debugging
-            return
+            return video_maker
 
-        self.insert1(key)
+        if output_video_filename.exists():
+            self.insert1(key)