Skip to content

Commit

Permalink
fix: use entry-point for density method in gpunufft.
Browse files Browse the repository at this point in the history
  • Loading branch information
paquiteau committed Jan 10, 2024
1 parent 95ab98e commit e6f5598
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions src/mrinufft/operators/interfaces/gpunufft.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import warnings
from ..base import FourierOperatorBase
from mrinufft._utils import proper_trajectory
from mrinufft.density import get_density

GPUNUFFT_AVAILABLE = True
try:
Expand Down Expand Up @@ -300,13 +301,7 @@ def __init__(
self.dtype = self.samples.dtype
self.n_coils = n_coils
self.smaps = smaps
if density is True:
self.density = self.pipe(self.samples, shape)
elif isinstance(density, np.ndarray):
self.density = density
else:
self.density = None
self.kwargs = kwargs
self.density = get_density(density)
self.impl = RawGpuNUFFT(
samples=self.samples,
shape=self.shape,
Expand Down

0 comments on commit e6f5598

Please sign in to comment.