jaxdem.rl.envWrapper

Contains wrappers for modifying rl environments.

Functions

clip_action_env(env[, min_val, max_val])	Wrap an environment so that its step method clips the action before calling the original step.
vectorise_env(env)	Promote an environment instance to a parallel version by applying jax.vmap(...) to its static methods.

jaxdem.rl.envWrapper.vectorise_env(env: Environment) → Environment

Promote an environment instance to a parallel version by applying jax.vmap(…) to its static methods.

jaxdem.rl.envWrapper.clip_action_env(env: Environment, min_val: float = -1.0, max_val: float = 1.0) → Environment

Wrap an environment so that its step method clips the action before calling the original step.