Hyde
/
yes_cmdr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

main
hkr04 2024-12-20 19:23:31 +08:00
parent 7a03d40d51
commit 9676d25ad6
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
yes_cmdr_env.py
View File

@ -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)