Fix resolution check. (#49)

e3nn · Dec 9, 2023 · 4405376 · 4405376
1 parent 0d5038e
commit 4405376
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/e3nn_jax/_src/s2grid.py b/e3nn_jax/_src/s2grid.py
@@ -912,9 +912,9 @@ def _from_s2grid_s2fft(
     """An S2FFT powered version of e3nn_jax.from_s2grid."""
     lmax = irreps.lmax
     expected_grid_resolution = get_s2fft_grid_resolution(lmax)
-    if sig.shape != expected_grid_resolution:
+    if sig.grid_resolution != expected_grid_resolution:
         raise ValueError(
-            f"Input signal shape {sig.shape} does not match the required shape {expected_grid_resolution}"
+            f"Input signal resolution {sig.grid_resolution} does not match the required resolution {expected_grid_resolution}."
         )
 
     with jax.ensure_compile_time_eval():