diff --git a/manual_control.py b/manual_control.py index b3eda90..79eae39 100755 --- a/manual_control.py +++ b/manual_control.py @@ -32,6 +32,8 @@ args = parser.parse_args() if args.env_name and args.env_name.find("Duckietown") != -1: + env = gym.make(args.env_name) +else: env = DuckietownEnv( seed=args.seed, map_name=args.map_name, @@ -43,8 +45,6 @@ camera_rand=args.camera_rand, dynamics_rand=args.dynamics_rand, ) -else: - env = gym.make(args.env_name) env.reset() env.render()