ding.envs.env_manager.ding_env_manager

ding.envs.env_manager.ding_env_manager

Full Source Code

../ding/envs/env_manager/ding_env_manager.py

1from . import BaseEnvManagerV2, SubprocessEnvManagerV2 2from ..env import DingEnvWrapper 3from typing import Optional 4from functools import partial 5 6 7def setup_ding_env_manager( 8 env: DingEnvWrapper, 9 env_num: int, 10 context: Optional[str] = None, 11 debug: bool = False, 12 caller: str = 'collector' 13) -> BaseEnvManagerV2: 14 assert caller in ['evaluator', 'collector'] 15 if debug: 16 env_cls = BaseEnvManagerV2 17 manager_cfg = env_cls.default_config() 18 else: 19 env_cls = SubprocessEnvManagerV2 20 manager_cfg = env_cls.default_config() 21 if context is not None: 22 manager_cfg.context = context 23 return env_cls([partial(env.clone, caller) for _ in range(env_num)], manager_cfg)