schola.sb3.action_space_patch.ActionSpacePatch

Class Definition

class schola.sb3.action_space_patch.ActionSpacePatch(
    globs,
    discrete_norm_factor=1.0,
    continuous_norm_factor=1.0
)

Bases: object

A context manager that patches the stable baselines3 library to support custom action spaces. This is done by overriding the make_proba_distribution function in the stable baselines3 library with a custom function that supports custom action spaces.

Note

Currently only works with PPO. Discrete and continuous actions are balanced with the discrete_norm_factor and continuous_norm_factor respectively.

Parameters

globs

Type: Dict
The globals dictionary of the calling module.

discrete_norm_factor

Type: float, optional
Default: 1.0
The normalization factor for discrete actions.

continuous_norm_factor

Type: float, optional
Default: 1.0
The normalization factor for continuous actions.

Methods

__init__

__init__(globs, discrete_norm_factor=1.0, continuous_norm_factor=1.0)