Available Environments

Environment Vectors

VecEnv

class pytorchrl.agent.env.vec_env.VecEnv[source]

Bases: object

Class to handle creation of environment vectors

classmethod create_factory(env_fn, env_kwargs={}, vec_env_size=1, log_dir=None, info_keywords=())[source]

Returns a function to create a vector of environments of size num_processes, so it can be executed by any worker, remote or not.

Parameters
  • env_fn (func) – Function to create the environment.

  • env_kwargs (dict) – keyword arguments of env_fn.

  • vec_env_size (int) – size of the vector of environments.

  • log_dir (str) – Target path for envs to log information through bench.Monitor class.

  • info_keywords (tuple) – Information keywords to be logged stored by bench.Monitor class.

Returns

  • make_vec_env (func) – Function to create a vector of environments.

  • dummy_env.action_space (gym.Space) – Environments action space.

  • dummy_env.observation_space (gym.Space) – Environments observation space.