def make_env(args):
assert (args.env in Robotics_envs_id or args.env in Kuka_envs_id)
if args.env[:4] == 'Kuka':
return kuka_env.make_env(args)
elif args.env[:5] == 'Fetch':
return fetch_env.make_env(args)
else:
return hand_env.make_env(args)
def make_env(args):
# assert args.env in Robotics_envs_id
if args.env == 'Mountaincar-v0':
return mountain_car.make_env(args)
if args.env == 'World-v0':
return world.make_env(args)
if args.env == 'Pendulum-v0':
return pendulum.make_env(args)
if args.env == 'AntLocomotion-v0':
return ant_locomotion.make_env(args)
if args.env == 'AntLocomotionDiverse-v0':
return ant_locomotion_diverse.make_env(args)
if args.env == 'HalfCheetahGoal-v0':
return half_cheetah.make_env(args)
if args.env == 'Reacher-v0':
return reacher.make_env(args)
if args.env[:5] == 'Fetch':
return fetch_env.make_env(args)
else: # Hand envs
return hand_env.make_env(args)
def make_env(args): assert args.env in Robotics_envs_id if args.env[:5]=='Fetch': return fetch_env.make_env(args) else: # Hand envs return hand_env.make_env(args)