robosuite.wrappers package#

Submodules#

robosuite.wrappers.data_collection_wrapper module#

robosuite.wrappers.demo_sampler_wrapper module#

robosuite.wrappers.domain_randomization_wrapper module#

robosuite.wrappers.gym_wrapper module#

robosuite.wrappers.visualization_wrapper module#

robosuite.wrappers.wrapper module#

This file contains the base wrapper class for Mujoco environments. Wrappers are useful for data collection and logging. Highly recommended.

class robosuite.wrappers.wrapper.Wrapper(env)#

Bases: object

Base class for all wrappers in robosuite.

Parameters:

env (MujocoEnv) – The environment to wrap.

property action_dim#

By default, grabs the normal environment action_dim

Returns:

Action space dimension

Return type:

int

property action_spec#

By default, grabs the normal environment action_spec

Returns:

  • (np.array) minimum (low) action values

  • (np.array) maximum (high) action values

Return type:

2-tuple

classmethod class_name()#
observation_spec()#

By default, grabs the normal environment observation_spec

Returns:

Observations from the environment

Return type:

OrderedDict

render(**kwargs)#

By default, run the normal environment render() function

Parameters:

**kwargs (dict) – Any args to pass to environment render function

reset()#

By default, run the normal environment reset() function

Returns:

Environment observation space after reset occurs

Return type:

OrderedDict

step(action)#

By default, run the normal environment step() function

Parameters:

action (np.array) – action to take in environment

Returns:

  • (OrderedDict) observations from the environment

  • (float) reward from the environment

  • (bool) whether the current episode is completed or not

  • (dict) misc information

Return type:

4-tuple

property unwrapped#

Grabs unwrapped environment

Returns:

Unwrapped environment

Return type:

env (MujocoEnv)

Module contents#