rofunc.learning.RofuncRL.utils.stb3_utils

1.  Module Contents

1.1.  Classes

StableBaseline3Wrapper

1.2.  API

class rofunc.learning.RofuncRL.utils.stb3_utils.StableBaseline3Wrapper(env: Any, auto_reset_after_done: bool = True)

Bases: stable_baselines3.common.vec_env.VecEnv

Initialization

Isaac Gym environment (preview 3) wrapper for StableBaseline3

Parameters:

env – The IsaacGym environment to wrap

step(actions: torch.Tensor) → Tuple[torch.Tensor, torch.Tensor, torch.Tensor, Any]

Perform a step in the environment

Parameters:

actions (torch.Tensor) – The actions to perform

Returns:

The state, the reward, the done flag, and the info

Return type:

tuple of torch.Tensor and any other info

reset(env_idxs=False) → torch.Tensor

Reset the environment

Returns:

The state of the environment

Return type:

torch.Tensor

render(*args, **kwargs) → None

Render the environment

close() → None

Close the environment

step_async(actions)

step_wait()

get_attr(attr_name, indices=None)

set_attr(attr_name, value, indices=None)

env_method(method_name, *method_args, indices=None, **method_kwargs)

seed(seed)

env_is_wrapped()