Remove old t2 functions, all tests/lints passing

flatironinstitute · Oct 13, 2023 · 3a6ea17 · 3a6ea17
1 parent da65bff
commit 3a6ea17
Show file tree

Hide file tree

Showing 9 changed files with 55 additions and 1,143 deletions.
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