Moved convolutions and exposed them directly

NVIDIA · Dec 17, 2023 · 53ccd15 · 53ccd15
1 parent 1e5f7a2
commit 53ccd15
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/torch_harmonics/__init__.py b/torch_harmonics/__init__.py
@@ -32,8 +32,7 @@
 __version__ = '0.6.4'
 
 from .sht import RealSHT, InverseRealSHT, RealVectorSHT, InverseRealVectorSHT
+from .convolution import DiscreteContinuousConvS2, DiscreteContinuousConvTransposeS2
 from . import quadrature
-from . import s2_convolutions
-from . import disco_convolutions
 from . import random_fields
 from . import examples
diff --git a/torch_harmonics/s2_convolutions.py → torch_harmonics/convolution.py b/torch_harmonics/s2_convolutions.py → torch_harmonics/convolution.py