fix: cleanup density

mind-inria · Jan 10, 2024 · 2f39bcf · 2f39bcf
1 parent d7233c7
commit 2f39bcf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mrinufft/operators/interfaces/gpunufft.py b/src/mrinufft/operators/interfaces/gpunufft.py
@@ -377,6 +377,6 @@ def pipe(cls, kspace_loc, volume_shape, num_iterations=10):
         density_comp = np.ones(kspace_loc.shape[0])
         for _ in range(num_iterations):
             density_comp = density_comp / np.abs(
-                grid_op.op(grid_op.adj_op(density_comp, True), True)
+                grid_op.op(grid_op.adj_op(density_comp, None, True), None, True)
             )
         return density_comp