From 01a2c834ce3c8ff23c90ff098ef2cce04bdd5bab Mon Sep 17 00:00:00 2001
From: Nick Todd <nick.todd@t-s.qinetiq.com>
Date: Fri, 13 Sep 2024 14:53:15 +0100
Subject: [PATCH] #2879: Write seed value to log file.

---
 src/primaite/session/environment.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/primaite/session/environment.py b/src/primaite/session/environment.py
index 0fd21b9f..9054106e 100644
--- a/src/primaite/session/environment.py
+++ b/src/primaite/session/environment.py
@@ -62,6 +62,13 @@ def set_random_seed(seed: int, generate_seed_value: bool) -> Union[None, int]:
     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):
     """
     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}")
 
+        log_seed_value(self.seed)
+
     def action_masks(self) -> np.ndarray:
         """
         Return the action mask for the agent.