Merge pull request #7 from abhijeetgangan/revert-6-Tersoff_unit_test

Revert "Try #2"
abhijeetgangan · May 22, 2024 · 0e26bea · 0e26bea
2 parents 40d8c27 + d2a7296
commit 0e26bea
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/rigid_body_test.py b/tests/rigid_body_test.py
@@ -28,10 +28,7 @@
 from jax import lax
 from jax import test_util as jtu
 
-
-from jax.config import config as jax_config
-jax_config.update('jax_disable_jit', True)
-
+from jax import config as jax_config
 import jax.numpy as jnp
 
 from jax_md import quantity