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

#2085: generate time based log files

Browse Source
This commit is contained in:
Nick Todd
2023-11-30 16:11:44 +00:00
parent 3ab911f6af
commit 5cd69f343f
2 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,4 +1,4 @@
import os
# import os
from typing import Any, Dict, Final, Optional, SupportsFloat, Tuple
import gymnasium
@@ -37,11 +37,11 @@ class PrimaiteGymEnv(gymnasium.Env):
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_steps_log_file_path, "a", encoding="utf-8") as f:
f.write(str(dump_state))
f.write("\n=================\n")
f.flush()
# 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()
self.game.update_agents(state)