#917 - Got things working'ish

2023-06-20 22:29:46 +01:00
parent 7b0f47d6f8
commit dce6fe55ee
3 changed files with 16 additions and 5 deletions
--- a/src/primaite/agents/sb3.py
+++ b/src/primaite/agents/sb3.py
@@ -86,10 +86,10 @@ class SB3Agent(AgentSessionABC):

        if not episodes:
            episodes = self._training_config.num_episodes
-
        for i in range(episodes):
            self._agent.learn(total_timesteps=time_steps)
            self._save_checkpoint()
+
        self._env.close()
        super().learn()