stable_gym.common

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

Submodules

• stable_gym.common.max_episode_steps_injection_wrapper
• stable_gym.common.utils

Package Contents

Classes

MaxEpisodeStepsInjectionWrapper

A gymnasium wrapper that injects the max_episode_steps attribute of the

class stable_gym.common.MaxEpisodeStepsInjectionWrapper(env)

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