Add environment variable to disable runtime exceptions

ami-iit · Dec 18, 2024 · faf62d9 · faf62d9
1 parent be7f69a
commit faf62d9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/jaxsim/exceptions.py b/src/jaxsim/exceptions.py
@@ -1,3 +1,5 @@
+import os
+
 import jax
 
 
@@ -18,7 +20,9 @@ def raise_if(
     """
 
     # Disable host callback if running on TPU.
-    if jax.devices()[0].platform == "tpu":
+    if jax.devices()[0].platform == "tpu" or os.environ.get(
+        "JAXSIM_DISABLE_EXCEPTIONS", 0
+    ):
         return
 
     # Check early that the format string is well-formed.