Merge pull request #81 from flatironinstitute/cleanup/remove-old-t2

Remove old type 2 code
flatironinstitute · Oct 13, 2023 · a9caabe · a9caabe
2 parents 92c5eaf + 3a6ea17
commit a9caabe
Show file tree

Hide file tree

Showing 11 changed files with 57 additions and 1,270 deletions.
diff --git a/pytorch_finufft/__init__.py b/pytorch_finufft/__init__.py
@@ -1,6 +1,6 @@
 """Pytorch bindings for the FINUFFT Library"""
 
-from . import functional, nn
+from . import functional
 
-__all__ = ["functional", "nn"]
+__all__ = ["functional"]
 __version__ = "0.1.0"
diff --git a/pytorch_finufft/checks.py b/pytorch_finufft/checks.py
@@ -87,9 +87,7 @@ def check_output_shape(ndim: int, output_shape: Union[int, Tuple[int, ...]]) ->
 _COORD_CHAR_TABLE = "xyz"
 
 
-def _type2_checks(
-    points_tuple: Tuple[torch.Tensor, ...], targets: torch.Tensor
-) -> None:
+def _type2_checks(points_tuple: torch.Tensor, targets: torch.Tensor) -> None:
     """
     Performs all type, precision, size, device, ... checks for the
     type 2 FINUFFT