Skip to content

Commit 0738bd2

Browse files
authored
Add missing walk task in Humanoid_CMU
(Tassa et al., 2020, page 25, third paragraph from bottom)[https://arxiv.org/abs/2006.12983] describes the Humanoid_CMU task as having the `stand`, `walk` and `run` tasks -- however the `walk` task is missing.
1 parent f4e5e23 commit 0738bd2

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

dm_control/suite/humanoid_CMU.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,15 @@ def stand(time_limit=_DEFAULT_TIME_LIMIT, random=None, environment_kwargs=None):
5454
physics, task, time_limit=time_limit, control_timestep=_CONTROL_TIMESTEP,
5555
**environment_kwargs)
5656

57+
@SUITE.add()
58+
def walk(time_limit=_DEFAULT_TIME_LIMIT, random=None, environment_kwargs=None):
59+
"""Returns the Walk task."""
60+
physics = Physics.from_xml_string(*get_model_and_assets())
61+
task = HumanoidCMU(move_speed=_WALK_SPEED, random=random)
62+
environment_kwargs = environment_kwargs or {}
63+
return control.Environment(
64+
physics, task, time_limit=time_limit, control_timestep=_CONTROL_TIMESTEP,
65+
**environment_kwargs)
5766

5867
@SUITE.add()
5968
def run(time_limit=_DEFAULT_TIME_LIMIT, random=None, environment_kwargs=None):

0 commit comments

Comments
 (0)