fix tests

Author: mariogeiger

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)

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])