kitae.envs package

Subpackages

• kitae.envs.wrappers package
  • Submodules
  • kitae.envs.wrappers.compatibility module
    • EnvpoolCompatibility
      • EnvpoolCompatibility.reset()
    • SubProcVecParallelEnvCompatibility
      • SubProcVecParallelEnvCompatibility.reset()
  • kitae.envs.wrappers.record_episode_statistics module
    • ParallelRecordEpisodeStatistics
      • ParallelRecordEpisodeStatistics.return_queue
      • ParallelRecordEpisodeStatistics.length_queue
      • ParallelRecordEpisodeStatistics.reset()
      • ParallelRecordEpisodeStatistics.step()
  • kitae.envs.wrappers.vector module
    • CloudPickleWrapper
    • SubProcVecParallelEnv
      • SubProcVecParallelEnv.close()
      • SubProcVecParallelEnv.reset()
      • SubProcVecParallelEnv.step_async()
      • SubProcVecParallelEnv.step_wait()
    • VecParallelEnv
      • VecParallelEnv.close()
      • VecParallelEnv.close_extras()
      • VecParallelEnv.get_images()
      • VecParallelEnv.get_wrapper_attr()
      • VecParallelEnv.render()
      • VecParallelEnv.reset()
      • VecParallelEnv.step()
      • VecParallelEnv.step_async()
      • VecParallelEnv.step_wait()
    • stack()
    • tile_images()
    • worker()
  • Module contents

Submodules

kitae.envs.make module

kitae.envs.make.make_env(env_id: str, idx: int, *, capture_video: bool, run_name: str, **env_kwargs) → Callable

kitae.envs.make.make_single_env(env_id: str, *, capture_video: bool, run_name: str, **env_kwargs) → SyncVectorEnv

kitae.envs.make.make_vec_env(env_id: str, n_envs: int, *, capture_video: bool, run_name: str, **env_kwargs) → AsyncVectorEnv

kitae.envs.make.wrap_multiple_parallel_env(env: ParallelEnv, n_envs: int) → SubProcVecParallelEnv

kitae.envs.make.wrap_single_env(env: Env) → SyncVectorEnv

kitae.envs.make.wrap_single_parallel_env(env: ParallelEnv) → SubProcVecParallelEnv

Module contents