Fix type

AllenInstitute · Sep 14, 2023 · e2e901b · e2e901b
1 parent 2c0656d
commit e2e901b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/npc_sessions/sessions.py b/src/npc_sessions/sessions.py
@@ -963,7 +963,7 @@ def stim_frame_times(self) -> dict[str, npt.NDArray[np.float64]]:
             }
         asserted_stim_frame_times: dict[str, npt.NDArray[np.float64]] = {}
         for k, v in self._stim_frame_times.items():
-            utils.assert_stim_times(self._stim_frame_times[k])
+            v = utils.assert_stim_times(v)
             asserted_stim_frame_times[k] = v
         assert not any(isinstance(v, Exception) for v in asserted_stim_frame_times.values())
         return asserted_stim_frame_times