Merge remote-tracking branch 'origin/dev' into feature/2085-dump_describe_state

src/primaite/session/environment.py

@@ -62,6 +62,10 @@ class PrimaiteGymEnv(gymnasium.Env):
 
     def reset(self, seed: Optional[int] = None) -> Tuple[ObsType, Dict[str, Any]]:
         """Reset the environment."""
+        print(
+            f"Resetting environment, episode {self.game.episode_counter}, "
+            f"avg. reward: {self.game.rl_agents[0].reward_function.total_reward}"
+        )
         self.game.reset()
         state = self.game.get_sim_state()
         self.game.update_agents(state)

src/primaite/session/session.py

@@ -62,6 +62,7 @@ class PrimaiteSession:
 
     def start_session(self) -> None:
         """Commence the training/eval session."""
+        print("Starting Primaite Session")
         self.mode = SessionMode.TRAIN
         n_learn_episodes = self.training_options.n_learn_episodes
         n_eval_episodes = self.training_options.n_eval_episodes