Also accept mujoco.* as the main tag in the parser.

DeepMind · alimuldal · commit 3fdb2af22495 · 2020-09-25T16:03:13.000+01:00
PiperOrigin-RevId: 331411703
Change-Id: I3821e325ca0ee76d060db45a4b5d9ff14c7348ce
diff --git a/dm_control/mjcf/parser.py b/dm_control/mjcf/parser.py
@@ -132,13 +132,13 @@ def _parse(xml_root, escape_separators=False,
     An `mjcf.RootElement`.
 
   Raises:
-    ValueError: If `xml_root`'s tag is not 'mujoco'.
+    ValueError: If `xml_root`'s tag is not 'mujoco.*'.
   """
 
   assets = assets or {}
 
-  if xml_root.tag != 'mujoco':
-    raise ValueError('Root element of the XML should be <mujoco>: got <{}>'
+  if not xml_root.tag.startswith('mujoco'):
+    raise ValueError('Root element of the XML should be <mujoco.*>: got <{}>'
                      .format(xml_root.tag))
 
   with debugging.freeze_current_stack_trace():
diff --git a/setup.py b/setup.py
@@ -177,7 +177,7 @@ def is_excluded(s):
 
 setup(
     name='dm_control',
-    version='0.0.330388914',
+    version='0.0.331411703',
     description='Continuous control environments and MuJoCo Python bindings.',
     author='DeepMind',
     license='Apache License, Version 2.0',
