Merge pull request #106 from NeuTTH/main

Return sampled instances in generate_grid_movies
dattalab · Oct 21, 2023 · 2645b14 · 2645b14
2 parents b1d1492 + 4f5826a
commit 2645b14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keypoint_moseq/viz.py b/keypoint_moseq/viz.py
@@ -1210,7 +1210,7 @@ def generate_grid_movies(
 
         path = os.path.join(output_dir, f"syllable{syllable}.mp4")
         write_video_clip(frames, path, fps=fps, quality=quality)
-
+    return sampled_instances
 
 def get_limits(
     coordinates,