fix tests

e3nn · Nov 6, 2023 · e366b3c · e366b3c
1 parent 22ed24d
commit e366b3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/e3nn_jax/_src/irreps_test.py b/e3nn_jax/_src/irreps_test.py
@@ -146,4 +146,4 @@ def test_D(keys, ir):
     w = e3nn.angles_to_log_coordinates(*angles)
     Dw = ir.D_from_log_coordinates(w)
 
-    np.testing.assert_allclose(Da, Dw, atol=1e-6)
+    np.testing.assert_allclose(Da, Dw, atol=1e-10, rtol=0.0008)
diff --git a/e3nn_jax/_src/spherical_harmonics/recursive.py b/e3nn_jax/_src/spherical_harmonics/recursive.py
@@ -15,8 +15,6 @@ def recursive_spherical_harmonics(
     normalization: str,
     algorithm: Tuple[str],
 ) -> sympy.Array:
-    context.update(dict(jnp=jnp, clebsch_gordan=e3nn.clebsch_gordan))
-
     if l == 0:
         if 0 not in context:
             if normalization == "integral":
@@ -37,7 +35,7 @@ def recursive_spherical_harmonics(
             elif normalization == "component":
                 context[1] = math.sqrt(3) * input
             else:
-                context[1] = input
+                context[1] = 1 * input
 
         return sympy.Array([1, 0, 0])