未验证 提交 6326529c 编写于 作者: Y Yinmin.Zhang 提交者: GitHub

fix(zym): modify default setting in mujoco (#35)

上级 020eba28
from typing import Dict
import gym
import numpy as np
from ding.envs import ObsNormEnv, RewardNormEnv
def wrap_mujoco(env_id, norm_obs=True, norm_reward=True, only_info=False) -> gym.Env:
def wrap_mujoco(env_id, norm_obs: Dict=dict(use_norm=False, ),
norm_reward: Dict=dict(use_norm=False, ), only_info=False) -> gym.Env:
r"""
Overview:
Wrap Mujoco Env to preprocess env step's return info, e.g. observation normalization, reward normalization, etc.
......@@ -28,5 +30,5 @@ def wrap_mujoco(env_id, norm_obs=True, norm_reward=True, only_info=False) -> gym
if norm_obs is not None and norm_obs.use_norm:
wrapper_info = ObsNormEnv.__name__ + '\n'
if norm_reward is not None and norm_reward.use_norm:
wrapper_info = RewardNormEnv.__name__ + '\n'
wrapper_info += RewardNormEnv.__name__ + '\n'
return wrapper_info
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册