num_drones=ARGS.num_drones,
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=ARGS.obs,
act=ARGS.act))
elif ARGS.env == 'leaderfollower':
register_env(
temp_env_name, lambda _: LeaderFollowerAviary(
num_drones=ARGS.num_drones,
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=ARGS.obs,
act=ARGS.act))
elif ARGS.env == 'meetup':
register_env(
temp_env_name, lambda _: MeetupAviary(
num_drones=ARGS.num_drones,
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=ARGS.obs,
act=ARGS.act))
elif ARGS.env == 'payloadcoop':
register_env(
temp_env_name, lambda _: PayloadCoop(
num_drones=ARGS.num_drones,
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=ARGS.obs,
act=ARGS.act))
else:
print("[ERROR] environment not yet implemented")
exit()
#### Unused env to extract the act and obs spaces ##########
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=OBS,
act=ACT
)
)
elif ARGS.exp.split("-")[1] == 'leaderfollower':
register_env(temp_env_name, lambda _: LeaderFollowerAviary(num_drones=NUM_DRONES,
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=OBS,
act=ACT
)
)
elif ARGS.exp.split("-")[1] == 'meetup':
register_env(temp_env_name, lambda _: MeetupAviary(num_drones=NUM_DRONES,
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=OBS,
act=ACT
)
)
else:
print("[ERROR] environment not yet implemented")
exit()
#### Unused env to extract the act and obs spaces ##########
if ARGS.exp.split("-")[1] == 'flock':
temp_env = FlockAviary(num_drones=NUM_DRONES,
aggregate_phy_steps=shared_constants.AGGR_PHY_STEPS,
obs=OBS,
act=ACT
)
elif ARGS.exp.split("-")[1] == 'leaderfollower':