Skip to content

lit_sb3.core.import_envs

source module lit_sb3.core.import_envs

Register optional third-party gym environments and no-velocity variants.

Adapted from https://github.com/DLR-RM/rl-baselines3-zoo/blob/master/rl_zoo3/import_envs.py

Optional environment packages are imported with graceful fallback so that missing packages do not break the rest of the package.

Functions

source create_no_vel_env(env_id: str) → Callable[[str | None], gym.Env]

Return a factory that creates env_id wrapped with :class:MaskVelocityWrapper.