Add restriction to fetch1_dataframe

CBroz1 · Sep 21, 2023 · 4e6b29f · 4e6b29f
1 parent f76666f
commit 4e6b29f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/spyglass/common/common_behav.py b/src/spyglass/common/common_behav.py
@@ -226,7 +226,11 @@ def fetch_nwb(self, *attrs, **kwargs) -> list:
         """
         Returns a condatenated list of nwb objects from RawPosition.PosObject
         """
-        return self.PosObject().fetch_nwb(*attrs, **kwargs)
+        return (
+            self.PosObject()
+            .restrict(self.restriction)  # Avoids fetch_nwb on whole table
+            .fetch_nwb(*attrs, **kwargs)
+        )
 
     def fetch1_dataframe(self):
         """Returns a dataframe with all RawPosition.PosObject items.