DI-engine/ding/envs/gym_env.py

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)