Alignment: fixed warning

Signed-off-by: Nicola VIGANO <nicola.vigano@esrf.fr>
cicwi · Mar 27, 2024 · d56bd8c · d56bd8c
1 parent 5447b2f
commit d56bd8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/corrct/alignment/centering.py b/corrct/alignment/centering.py
@@ -58,8 +58,8 @@ def to_com(self, shifts_vu: Union[ArrayLike, NDArray], volume: NDArray, com_ref_
             The corrected VU shifts.
         """
         com_rec_zyx = post_proc.com(volume)
-        displacemenet_zyx = com_ref_zyx - com_rec_zyx
-        return self._apply_displacement_vu(shifts_vu, displacemenet_zyx)
+        displacemenet_zyx = np.array(com_ref_zyx) - com_rec_zyx
+        return self._apply_displacement_vu(np.array(shifts_vu), displacemenet_zyx)
 
     def as_reference(self, shifts_vu: NDArray, volume: NDArray, reference: NDArray, method: str = "com") -> NDArray:
         """Recenter with respect to a given volume.