fix(stacked): typo

mind-inria · Oct 16, 2023 · d7f1261 · d7f1261
1 parent 510ce19
commit d7f1261
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/mrinufft/operators/stacked.py b/src/mrinufft/operators/stacked.py
@@ -385,7 +385,7 @@ def _op_sense_host(self, data, ksp):
                 cp.copyto(coil_img_d, self.smaps[idx_coils])
             coil_img_d *= data_batched
             # FFT along Z axis (last)
-            coil_img_d = self.fftz(coil_img_d)
+            coil_img_d = self._fftz(coil_img_d)
             coil_img_d = coil_img_d.reshape((T, *XYZ))
             tmp = coil_img_d[..., self.z_index]
             tmp = cp.moveaxis(tmp, -1, 1)
@@ -414,7 +414,7 @@ def _op_calibless_host(self, data, ksp=None):
 
         for i in range((B * C) // T):
             coil_img_d.set(dataf[i * T : (i + 1) * T])
-            coil_img_d = self.fftz(coil_img_d)
+            coil_img_d = self._fftz(coil_img_d)
             coil_img_d = coil_img_d.reshape((T, *XYZ))
             tmp = coil_img_d[..., self.z_index]
             tmp = cp.moveaxis(tmp, -1, 1)
@@ -472,7 +472,7 @@ def _adj_op_sense_host(self, coeffs, img):
             tmp_adj = cp.moveaxis(tmp_adj, 1, -1)
             coil_img_d = cp.zeros_like(coil_img_d)
             coil_img_d[..., self.z_index] = tmp_adj
-            coil_img_d = self.ifftz(coil_img_d)
+            coil_img_d = self._ifftz(coil_img_d)
 
             for t, b in enumerate(idx_batch):
                 img_d[b, :] += coil_img_d[t] * smaps_batched[t].conj()
@@ -500,7 +500,7 @@ def _adj_op_calibless_host(self, coeffs, img):
 
             coil_img_d = cp.zeros_like(coil_img_d)
             coil_img_d[..., self.z_index] = tmp_adj
-            coil_img_d = self.ifftz(coil_img_d)
+            coil_img_d = self._ifftz(coil_img_d)
             img[i * T : (i + 1) * T] = coil_img_d.get()
         img = img.reshape(B, C, *XYZ)
         return img