def get_env_from_name(name, ENV_SEED=None):
if name == "oscillator":
from envs.oscillator import oscillator as env
env = env()
env = env.unwrapped
elif name == "cart_pole_cost":
from ENV.env.classic_control.ENV_V1 import CartPoleEnv_cost as env
env = env()
env = env.unwrapped
elif name == "Ex3_EKF":
from envs.Ex3_EKF import Ex3_EKF as env
env = env()
env = env.unwrapped
elif name == "Ex3_EKF_gyro":
from envs.Ex3_EKF_gyro import Ex3_EKF as env
env = env()
env = env.unwrapped
elif name == "Ex4_EKF":
from envs.Ex4_EKF import Ex4_EKF as env
env = env()
env = env.unwrapped
if ENV_SEED is not None:
env.seed(ENV_SEED)
return env
def get_env_from_name(name):
if name == 'cartpole_cost':
from envs.ENV_V1 import CartPoleEnv_adv as dreamer
env = dreamer()
env = env.unwrapped
elif name == 'cartpole_cost_v2':
from envs.ENV_V2 import CartPoleEnv_adv as dreamer
env = dreamer()
env = env.unwrapped
elif name == 'cartpole_cost_partial':
from envs.ENV_V3 import CartPoleEnv_adv as dreamer
env = dreamer()
env = env.unwrapped
elif name == 'cartpole_cost_real':
from envs.ENV_V4 import CartPoleEnv_adv as dreamer
env = dreamer()
env = env.unwrapped
elif name == 'cartpole_cost_swing_up':
from envs.ENV_V5 import CartPoleEnv_adv as dreamer
env = dreamer()
env = env.unwrapped
elif name == 'cartpole_cost_real_no_friction':
from envs.ENV_V6 import CartPoleEnv_adv as dreamer
env = dreamer()
env = env.unwrapped
elif name == 'cartpole_cost_with_motor':
from envs.ENV_V7 import CartPoleEnv_adv as dreamer
env = dreamer()
env = env.unwrapped
elif name == 'cartpole_cost_with_fitted_motor':
from envs.ENV_V8 import CartPoleEnv_adv as dreamer
env = dreamer(eval=True)
env = env.unwrapped
elif name == 'oscillator':
from envs.oscillator import oscillator as env
env = env()
env = env.unwrapped
elif name == 'oscillator_complicated':
from envs.oscillator_complicated import oscillator as env
env = env()
env = env.unwrapped
elif name == 'Quadrotorcost-v0':
env = gym.make('Quadrotorcons-v0')
env = env.unwrapped
env.modify_action_scale = False
env.use_cost = True
else:
env = gym.make(name)
env = env.unwrapped
if name == 'Quadrotorcons-v0':
if 'CPO' not in VARIANT['algorithm_name']:
env.modify_action_scale = False
if 'Fetch' in name or 'Hand' in name:
env.unwrapped.reward_type = 'dense'
env.seed(SEED)
return env
def get_env_from_name(name):
if name == 'CMAPSS':
from envs.CMAPSS import CMAPSS as env
env = env()
env = env.unwrapped
elif name == 'BatteryCalib':
from envs.BatteryCalib import BatteryCalib as env
env = env(action_low=ENV_PARAMS[name]['action_low'],
action_high=ENV_PARAMS[name]['action_high'])
env = env.unwrapped
else:
from envs.Battery_test import BatteryCalib as env
env = env(action_low=ENV_PARAMS[name]['action_low'],
action_high=ENV_PARAMS[name]['action_high'])
env = env.unwrapped
env.seed(SEED)
return env
def get_env_from_name(name, ENV_SEED=None):
if name == "oscillator":
from envs.oscillator import oscillator as env
env = env()
env = env.unwrapped
elif name == "Ex3_EKF":
from envs.Ex3_EKF import Ex3_EKF as env
env = env()
env = env.unwrapped
elif name == "Ex3_EKF_gyro":
from envs.Ex3_EKF_gyro import Ex3_EKF as env
env = env()
env = env.unwrapped
elif name == "Ex4_EKF":
from envs.Ex4_EKF import Ex4_EKF as env
env = env()
env = env.unwrapped
if ENV_SEED is not None:
env.seed(ENV_SEED)
return env