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)