fixing more issues from merge

f72a48dd · Boris Bonev · 828f2ba3 · f72a48dd
Commit f72a48dd authored Jul 21, 2025 by Boris Bonev
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 8 deletions

torch_harmonics/distributed/distributed_convolution.py torch_harmonics/distributed/distributed_convolution.py +3 -8

No files found.
--- a/torch_harmonics/distributed/distributed_convolution.py
+++ b/torch_harmonics/distributed/distributed_convolution.py
@@ -100,11 +100,6 @@ def _split_distributed_convolution_tensor_s2(
        Filtered values corresponding to the local latitude slice
    """

-    assert len(in_shape) == 2
-    assert len(out_shape) == 2
-
-    kernel_size = filter_basis.kernel_size
-
    nlat_in, nlon_in = in_shape
    nlat_out, nlon_out = out_shape