oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2085: Get enum value data

Browse Source
This commit is contained in:
Nick Todd
2023-12-01 09:52:31 +00:00
parent 10d4b61c73
commit a073038ec0
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/primaite/session/environment.py
View File

@@ -1,4 +1,4 @@
# import os
import json
from typing import Any, Dict, Final, Optional, SupportsFloat, Tuple
import gymnasium
@@ -34,14 +34,15 @@ class PrimaiteGymEnv(gymnasium.Env):
state = self.game.get_sim_state()
# Create state suitable for dumping to file.
dump_state = {self.game.episode_counter: {self.game.step_counter: state}}
# dump_state = {self.game.episode_counter: {self.game.step_counter: state}}
# Dump to file
# if os.path.isfile(PRIMAITE_PATHS.episode_steps_log_file_path):
with open(PRIMAITE_PATHS.episode_log_file_path, "a", encoding="utf-8") as f:
f.write(str(dump_state))
f.write("\n=================\n")
f.flush()
# f.write(str(dump_state))
# f.write("\n=================\n")
# f.flush()
json.dump(state, f)
self.game.update_agents(state)