diff --git a/yes_cmdr_env.py b/yes_cmdr_env.py index ae71475..79a14fd 100644 --- a/yes_cmdr_env.py +++ b/yes_cmdr_env.py @@ -1307,11 +1307,14 @@ class YesCmdrEnv(gym.Env): agent_ids = self.current_agents_dict.keys() return json_to_markdown(self.get_agents_dict(agent_ids)) - def format_battlefeild(self, battlefield_info): + def format_battlefields(self, battlefield_infos): dict_info = {} - dict_info["battlefield_name"] = battlefield_info["sectorName"] - dict_info["my_agents"] = self.get_agents_dict(battlefield_info["myUnitsID"]) - dict_info["ally_agents"] = self.get_agents_dict(battlefield_info["allyUnitsID"]) - dict_info["enemy_agents"] = self.get_agents_dict(battlefield_info["hostileUnitsID"]) + for battlefield_info in battlefield_infos: + _dict_info = {} + _dict_info["battlefield_name"] = battlefield_info["sectorName"] + _dict_info["my_agents"] = self.get_agents_dict(battlefield_info["myUnitsID"]) + _dict_info["ally_agents"] = self.get_agents_dict(battlefield_info["allyUnitsID"]) + _dict_info["enemy_agents"] = self.get_agents_dict(battlefield_info["hostileUnitsID"]) + dict_info[battlefield_info["sectorName"]] = _dict_info return json_to_markdown(dict_info) \ No newline at end of file