stable_gym.common

Contains several functions that can be used across all the Stable Gym gymnasium environments.

Submodules

Classes

MaxEpisodeStepsInjectionWrapper

A gymnasium wrapper that injects the max_episode_steps attribute of the

Package Contents

class stable_gym.common.MaxEpisodeStepsInjectionWrapper(env)[source]

Bases: gymnasium.Wrapper

A gymnasium wrapper that injects the max_episode_steps attribute of the TimeLimit wrapper into the base environment as the _time_limit_max_episode_steps attribute. If the environment is not wrapped in a TimeLimit wrapper, then the _time_limit_max_episode_steps attribute is set to None.

Wrap a gymnasium environment. :param env: The gymnasium environment. :type env: gym.Env

max_episode_steps