schola.sb3.action_space_patch

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

Functions

make_hybrid_dist(action_space[, use_sde, …])	Create a hybrid distribution from a dictionary of action spaces.
patched_get_action_dim(action_space)
patched_with_norm([discrete_norm_factor, …])
reshape_nonbatch(tensor)

Classes

ActionSpacePatch(globs[, …])	A context manager that patches the stable baselines3 library to support custom action spaces.
HybridDistribution(distributions[, …])	A composite distribution supporting discrete and continuous sub-distributions.
PatchedPPO(policy, env[, learning_rate, …])