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

#2879: Write seed value to log file.

Browse Source
This commit is contained in:
Nick Todd
2024-09-13 14:53:15 +01:00
parent 08fcf1df19
commit 01a2c834ce
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -62,6 +62,13 @@ def set_random_seed(seed: int, generate_seed_value: bool) -> Union[None, int]:
return seed return seed
def log_seed_value(seed: int):
"""Log the selected seed value to file."""
path = SIM_OUTPUT.path / "seed.log"
with open(path, "w") as file:
file.write(f"Seed value = {seed}")
class PrimaiteGymEnv(gymnasium.Env): class PrimaiteGymEnv(gymnasium.Env):
""" """
Thin wrapper env to provide agents with a gymnasium API. Thin wrapper env to provide agents with a gymnasium API.
@@ -92,6 +99,8 @@ class PrimaiteGymEnv(gymnasium.Env):
_LOGGER.info(f"PrimaiteGymEnv RNG seed = {self.seed}") _LOGGER.info(f"PrimaiteGymEnv RNG seed = {self.seed}")
log_seed_value(self.seed)
def action_masks(self) -> np.ndarray: def action_masks(self) -> np.ndarray:
""" """
Return the action mask for the agent. Return the action mask for the agent.