Update rigid_body_test.py

abhijeetgangan · May 22, 2024 · e20bffa · e20bffa
1 parent 90065ff
commit e20bffa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/rigid_body_test.py b/tests/rigid_body_test.py
@@ -29,6 +29,8 @@
 from jax import test_util as jtu
 
 from jax import config as jax_config
+jax_config.update('jax_disable_jit', True)
+
 import jax.numpy as jnp
 
 from jax_md import quantity