From da2fad4530eefa4ed269dc39a8a38af2c8d58693 Mon Sep 17 00:00:00 2001
From: Google DeepMind <noreply@deepmind.com>
Date: Wed, 10 Sep 2025 17:52:50 -0700
Subject: [PATCH] Catch RuntimeErrors when importing warp.

PiperOrigin-RevId: 805586914
Change-Id: Ica3ca016c8056316d1d2f052e4d3481f578a8aad
---
 mjx/mujoco/mjx/warp/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/mjx/mujoco/mjx/warp/__init__.py b/mjx/mujoco/mjx/warp/__init__.py
index 067adf1e..98793aef 100644
--- a/mjx/mujoco/mjx/warp/__init__.py
+++ b/mjx/mujoco/mjx/warp/__init__.py
@@ -28,12 +28,14 @@ if not typing.TYPE_CHECKING:
   try:
     import warp
     WARP_INSTALLED = True
-  except ImportError:
+  except (ImportError, RuntimeError) as e:
+    print('Failed to import warp:', e)
     WARP_INSTALLED = False
   try:
     import mujoco.mjx.third_party.mujoco_warp as mujoco_warp
     from mujoco.mjx.third_party.mujoco_warp._src import types as mjwp_types
-  except ImportError as e:
+  except (ImportError, RuntimeError) as e:
+    print('Failed to import mujoco.mjx.third_party.mujoco_warp as mujoco_warp:', e)
     pass
   # pylint: enable=g-import-not-at-top
 else: