Function balance_sparse

dm_control/suite/cartpole.py:51–58

Returns the sparse reward variant of the Cartpole Balance task.

(time_limit=_DEFAULT_TIME_LIMIT, random=None,
                   environment_kwargs=None)

@SUITE.add('benchmarking')
def balance_sparse(time_limit=_DEFAULT_TIME_LIMIT, random=None,
                   environment_kwargs=None):
  """Returns the sparse reward variant of the Cartpole Balance task."""
  physics = Physics.from_xml_string(*get_model_and_assets())
  task = Balance(swing_up=False, sparse=True, random=random)
  environment_kwargs = environment_kwargs or {}
  return control.Environment(
      physics, task, time_limit=time_limit, **environment_kwargs)


@SUITE.add('benchmarking')