fix: rewrite the off-resonance operator.

Author: paquiteau

src/mrinufft/operators/base.py

@@ -232,11 +232,20 @@ def data_consistency(self, image_data, obs_data):
         """
         return self.adj_op(self.op(image_data) - obs_data)
 
-    def with_off_resonance_correction(self, B, C, indices):
+    def with_off_resonance_correction(
+        self,
+        b0_map: NDArray | None = None,
+        readout_time: NDArray | None = None,
+        r2star_map: NDArray | None = None,
+        mask: NDArray | None = None,
+        interpolator: str | dict | tuple[NDArray, NDArray] = "svd",
+    ):
         """Return a new operator with Off Resonnance Correction."""
         from .off_resonance import MRIFourierCorrected
 
-        return MRIFourierCorrected(self, B, C, indices)
+        return MRIFourierCorrected(
+            self, b0_map, readout_time, r2star_map, mask, interpolator
+        )
 
     def compute_smaps(self, method: NDArray | Callable | str | dict | None = None):
         """Compute the sensitivity maps and set it.