habitat.core.vector_env module

Contents

Classes

class ThreadedVectorEnv
Provides same functionality as VectorEnv, the only difference is it runs in a multi-thread setup inside a single process.
class VectorEnv
Vectorized environment which creates multiple processes where each process runs its own environment. Main class for parallelization of training and evaluation.

Functions

def _make_env_fn(config: habitat.config.default.Config, dataset: typing.Optional[dataset.Dataset] = None, rank: int = 0) -> env.Env
Constructor for default habitat env.Env.

Data

ACTION_SPACE_COMMAND = 'action_space'
CALL_COMMAND = 'call'
CLOSE_COMMAND = 'close'
COUNT_EPISODES_COMMAND = 'count_episodes'
EPISODE_COMMAND = 'current_episode'
EPISODE_OVER = 'episode_over'
GET_METRICS = 'get_metrics'
NUMBER_OF_EPISODES_COMMAND = 'number_of_episodes'
OBSERVATION_SPACE_COMMAND = 'observation_space'
RENDER_COMMAND = 'render'
RESET_COMMAND = 'reset'
STEP_COMMAND = 'step'

Function documentation

def habitat.core.vector_env._make_env_fn(config: habitat.config.default.Config, dataset: typing.Optional[dataset.Dataset] = None, rank: int = 0) -> env.Env

Constructor for default habitat env.Env.

Parameters
config configuration for environment.
dataset dataset for environment.
rank rank for setting seed of environment
Returns env.Env / env.RLEnv object