Fix type annotation for get_shifts

nel-lab · Sep 8, 2024 · 7ccfee0 · 7ccfee0
1 parent 9b6f5dd
commit 7ccfee0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mesmerize_core/caiman_extensions/mcorr.py b/mesmerize_core/caiman_extensions/mcorr.py
@@ -91,7 +91,7 @@ def get_output(self, mode: str = "r") -> np.ndarray:
  return mc_movie
 
  @validate("mcorr")
- def get_shifts(self, pw_rigid) -> tuple[list[np.ndarray], list[np.ndarray]]:
+ def get_shifts(self, pw_rigid) -> list[np.ndarray]:
  """
  Gets file path to shifts array (.npy file) for item, processes shifts array
  into a list of x and y shifts based on whether rigid or nonrigid