diff --git a/gym_art/quadrotor_multi/quad_scenarios.py b/gym_art/quadrotor_multi/quad_scenarios.py index 9f9eb59..b955f9e 100644 --- a/gym_art/quadrotor_multi/quad_scenarios.py +++ b/gym_art/quadrotor_multi/quad_scenarios.py @@ -4,7 +4,7 @@ import copy from gym_art.quadrotor_multi.quad_scenarios_utils import QUADS_PARAMS_DICT, update_formation_and_max_agent_per_layer, \ update_layer_dist, get_formation_range, get_goal_by_formation, get_z_value, QUADS_MODE_LIST, \ - QUADS_MODE_LIST_OBSTACLES + QUADS_MODE_LIST_OBSTACLES, QUADS_MODE_LIST_SIMPLE from gym_art.quadrotor_multi.quad_utils import generate_points, get_grid_dim_number @@ -623,7 +623,9 @@ class Scenario_mix(QuadrotorScenario): # Once change the parameter here, should also update QUADS_PARAMS_DICT to make sure it is same as run a single scenario # key: quads_mode # value: 0. formation, 1: [formation_low_size, formation_high_size], 2: episode_time - if obst_mode == 'no_obstacles': + if num_agents == 1: + self.quads_mode_list = QUADS_MODE_LIST_SIMPLE + elif obst_mode == 'no_obstacles': self.quads_mode_list = QUADS_MODE_LIST else: self.quads_mode_list = QUADS_MODE_LIST_OBSTACLES diff --git a/gym_art/quadrotor_multi/quad_utils.py b/gym_art/quadrotor_multi/quad_utils.py index d56c5d5..a8792fc 100644 --- a/gym_art/quadrotor_multi/quad_utils.py +++ b/gym_art/quadrotor_multi/quad_utils.py @@ -208,7 +208,7 @@ def points_in_sphere(n, x): def generate_points(n=3): if n < 3: - print("The number of goals can not smaller than 3, The system has cast it to 3") + # print("The number of goals can not smaller than 3, The system has cast it to 3") n = 3 return points_in_sphere(n, 0.1 + 1.2 * n)