format and add str check

vikashplus · Dec 20, 2023 · 94fe08e · 94fe08e
1 parent 81e90cb
commit 94fe08e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/robohive/physics/mj_sim_scene.py b/robohive/physics/mj_sim_scene.py
@@ -37,8 +37,8 @@ def _load_simulation(self, model_handle: Any) -> Any:
         if isinstance(model_handle, str):
             if model_handle.endswith('.xml'):
                 sim = dm_mujoco.Physics.from_xml_path(model_handle)
-            elif "<mujoco" in model_handle:
-                            sim = dm_mujoco.Physics.from_xml_string(model_handle)
+            elif isinstance(model_handle, str) and "<mujoco" in model_handle:
+                sim = dm_mujoco.Physics.from_xml_string(model_handle)
             else:
                 sim = dm_mujoco.Physics.from_binary_path(model_handle)
         else: