Skip to content

Commit bc4b064

Browse files
dm_control: Import of refs/pull/336/head
PiperOrigin-RevId: 507528987 Change-Id: I76d0174749de9f87b0f842ec79648359a7c555f8
2 parents fdb8fe1 + 0738bd2 commit bc4b064

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

dm_control/suite/humanoid_CMU.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,17 @@ def stand(time_limit=_DEFAULT_TIME_LIMIT, random=None, environment_kwargs=None):
5555
**environment_kwargs)
5656

5757

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

0 commit comments

Comments
 (0)