Update viewer.py to work with MuJoCo 2.3.7.

google-research · Aug 3, 2023 · a0e13d4 · a0e13d4
1 parent f9d074f
commit a0e13d4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/robopianist/viewer/viewer.py b/robopianist/viewer/viewer.py
@@ -70,7 +70,10 @@
 _NEXT_LABELING_MODE = user_input.KEY_F7
 _PRINT_CAMERA = user_input.KEY_F11
 _VISUALIZATION_FLAGS = user_input.Range(
-    [ord(functions.mjVISSTRING[i][2]) for i in range(0, mujoco.mjtVisFlag.mjNVISFLAG)]
+    [
+        ord(functions.mjVISSTRING[i][2]) if functions.mjVISSTRING[i][2] else 0
+        for i in range(0, mujoco.mjtVisFlag.mjNVISFLAG)
+    ]
 )
 _GEOM_GROUPS = user_input.Range(
     [i + ord("0") for i in range(min(_NUM_GROUP_KEYS, mujoco.mjNGROUP))]