Remove some relative imports (#102)

mmuckley · Dec 4, 2024 · 8f8736c · 8f8736c
1 parent 399eabb
commit 8f8736c
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/torchkbnufft/functional/interp.py b/torchkbnufft/functional/interp.py
@@ -3,7 +3,7 @@
 import torch
 from torch import Tensor
 
-from .._autograd.interp import (
+from torchkbnufft._autograd.interp import (
     KbSpmatInterpAdjoint,
     KbSpmatInterpForward,
     KbTableInterpAdjoint,

diff --git a/torchkbnufft/functional/nufft.py b/torchkbnufft/functional/nufft.py
@@ -3,8 +3,8 @@
 import torch
 from torch import Tensor
 
-from .._nufft.fft import fft_and_scale, ifft_and_scale
-from .interp import (
+from torchkbnufft._nufft.fft import fft_and_scale, ifft_and_scale
+from torchkbnufft.functional.interp import (
     kb_spmat_interp,
     kb_spmat_interp_adjoint,
     kb_table_interp,

diff --git a/torchkbnufft/modules/_kbmodule.py b/torchkbnufft/modules/_kbmodule.py
@@ -3,7 +3,7 @@
 import torch
 import torch.nn as nn
 
-from .._nufft.utils import DTYPE_MAP, init_fn
+from torchkbnufft._nufft.utils import DTYPE_MAP, init_fn
 
 
 class KbModule(nn.Module):

diff --git a/torchkbnufft/modules/kbinterp.py b/torchkbnufft/modules/kbinterp.py
@@ -4,7 +4,7 @@
 import torchkbnufft.functional as tkbnF
 from torch import Tensor
 
-from ._kbmodule import KbModule
+from torchkbnufft.modules._kbmodule import KbModule
 
 
 class KbInterpModule(KbModule):

diff --git a/torchkbnufft/modules/kbnufft.py b/torchkbnufft/modules/kbnufft.py
@@ -4,8 +4,8 @@
 import torchkbnufft.functional as tkbnF
 from torch import Tensor
 
-from .._nufft.utils import compute_scaling_coefs
-from ._kbmodule import KbModule
+from torchkbnufft._nufft.utils import compute_scaling_coefs
+from torchkbnufft.modules._kbmodule import KbModule
 
 
 class KbNufftModule(KbModule):