Skip to content

schola.sb3.Action_space_patch

Tools for adapting Stable Baselines 3 PPO implementation to work with dictionary action spaces.

Functions

make_hybrid_distCreate a hybrid distribution from a dictionary of action spaces.
patched_get_action_dim
patched_with_norm
reshape_nonbatch

Classes

ActionSpacePatchA context manager that patches the stable baselines3 library to support custom action spaces.
HybridDistributionA composite distribution supporting discrete and continuous sub-distributions.
PatchedPPO