Environment_wrapper

conformer_rl.environments.environment_wrapper.Task(name: str, concurrency: bool = False, num_envs: int = 1, seed: int = 8084, **kwargs) → Union[stable_baselines3.common.vec_env.subproc_vec_env.SubprocVecEnv, conformer_rl.environments.simple_vec_env.SimpleVecEnv]

Returns a wrapper for wrapping multiple environments.

Parameters

  • name (str) – The name of the environment, as registered using the gym.register method.

  • concurrency (bool) – Whether or not the environments should be run in parallel across multiple CPU’s.

  • num_envs (bool) – The number of environments to be wrapped.

  • seed (bool) – Seed for initializing the environments.

Return type

A wrapper for the environment(s).