def _make_env_fn(config: Config, dataset: Optional[habitat.Dataset] = None, rank: int = 0) -> Env: """Constructor for default habitat `env.Env`. :param config: configuration for environment. :param dataset: dataset for environment. :param rank: rank for setting seed of environment :return: `env.Env` / `env.RLEnv` object """ habitat_env = Env(config=config, dataset=dataset) habitat_env.seed(config.SEED + rank) return habitat_env
def _make_env_fn(config: Config, dataset: Optional[habitat.Dataset] = None, rank: int = 0) -> Env: r"""Constructor for default habitat Env. Args: config: configuration for environment. dataset: dataset for environment. rank: rank for setting seed of environment Returns: ``Env``/``RLEnv`` object """ habitat_env = Env(config=config, dataset=dataset) habitat_env.seed(config.SEED + rank) return habitat_env