unilab.base.registry.env

unilab.base.registry.env(name, sim_backend)[source]

Decorator to register an environment class or function factory.

Usage:

@env(“my-env”, “mujoco”) class MyEnv(ABEnv):

…

@env(“my-manager-env”, “mujoco”) def make_my_env(cfg, num_envs=1, backend_type=”mujoco”):

…

Parameters:

  • name (str)

  • sim_backend (str)

Return type:

Callable[[TypeVar(TEnvFactory, bound= EnvFactory)], TypeVar(TEnvFactory, bound= EnvFactory)]