modify interval_list_name call

CBroz1 · Jul 27, 2023 · c228b82 · c228b82
1 parent 55e5261
commit c228b82
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/spyglass/position/v1/position_dlc_pose_estimation.py b/src/spyglass/position/v1/position_dlc_pose_estimation.py
@@ -234,7 +234,13 @@ def make(self, key):
             ).strftime("%Y-%m-%d %H:%M:%S")
 
             logger.logger.info("getting raw position")
-            interval_list_name = f"pos {key['epoch']-1} valid times"
+            interval_list_name = (
+                key["interval_list_name"]
+                if "interval_list_name" in key
+                else f"pos {key['epoch']-1} valid times"
+            )
+            if "interval_list_name" in key:
+                del key["interval_list_name"]
             spatial_series = (
                 RawPosition()
                 & {**key, "interval_list_name": interval_list_name}