File size: 234 Bytes
079c32c
 
 
 
 
 
1
2
3
4
5
6
7
from ding.envs import BaseEnv, DingEnvWrapper


def env(cfg, seed_api=True, caller='collector', **kwargs) -> BaseEnv:
    import gym
    return DingEnvWrapper(gym.make(cfg.env_id, **kwargs), cfg=cfg, seed_api=seed_api, caller=caller)