Use correct kernel construction functions

astro-informatics · Dec 11, 2024 · 8919b2f · 8919b2f
1 parent 8af3dd0
commit 8919b2f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/s2fft/precompute_transforms/spherical.py b/s2fft/precompute_transforms/spherical.py
@@ -624,7 +624,7 @@ def forward_transform_torch(
 }
 
 _kernel_functions = {
-    "numpy": partial(construct.fourier_wigner_kernel, using_torch=False),
-    "jax": construct.fourier_wigner_kernel_jax,
-    "torch": partial(construct.fourier_wigner_kernel, using_torch=True),
+    "numpy": partial(construct.spin_spherical_kernel, using_torch=False),
+    "jax": construct.spin_spherical_kernel_jax,
+    "torch": partial(construct.spin_spherical_kernel, using_torch=True),
 }