From 9676d25ad6f3312a5336282fb261b71d1f9f5f4f Mon Sep 17 00:00:00 2001 From: hkr04 Date: Fri, 20 Dec 2024 19:23:31 +0800 Subject: [PATCH] update --- yes_cmdr_env.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) 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