accept key in all get video path functions

LorenFrankLab · Jan 16, 2025 · 31014f9 · 31014f9
1 parent 2bff705
commit 31014f9
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/src/spyglass/common/common_position.py b/src/spyglass/common/common_position.py
@@ -514,8 +514,9 @@ def _data_to_df(
     def fetch_pose_datframe(self):
         raise NotImplementedError("No Pose data available for this table")
 
-    def fetch_video_path(self):
-        return self.fetch_nwb()[0]["head_position"].get_comments()
+    def fetch_video_path(self, key=dict()):
+        key = (self & key).fetch1("KEY")
+        return (self & key).fetch_nwb()[0]["head_position"].get_comments()
 
 
 @schema

diff --git a/src/spyglass/position/position_merge.py b/src/spyglass/position/position_merge.py
@@ -88,8 +88,8 @@ def fetch_pose_dataframe(self):
         )
         return query.fetch_pose_dataframe()
 
-    def fetch_video_path(self):
-        key = self.merge_restrict(self.proj()).proj()
+    def fetch_video_path(self, key=dict()):
+        key = self.merge_restrict((self & key).proj()).proj()
         query = (
             source_class_dict[
                 to_camel_case(self.merge_get_parent(self.proj()).table_name)

diff --git a/src/spyglass/position/v1/position_trodes_position.py b/src/spyglass/position/v1/position_trodes_position.py
@@ -254,7 +254,9 @@ def fetch1_dataframe(self, add_frame_ind=True) -> DataFrame:
     def fetch_pose_dataframe(self):
         raise NotImplementedError("No pose data for TrodesPosV1")
 
-    def fetch_video_path(self):
+    def fetch_video_path(self, key=dict()):
+        """Fetch the video path for the position data."""
+        key = (self & key).fetch1("KEY")
         nwb_file_name, interval_list_name = self.fetch1(
             "nwb_file_name", "interval_list_name"
         )