#1386: fix saving of agent

2023-07-05 11:41:18 +01:00
parent 818d64f330
commit b0c83d7148
3 changed files with 12 additions and 11 deletions
--- a/src/primaite/agents/rllib.py
+++ b/src/primaite/agents/rllib.py
@@ -141,8 +141,11 @@ class RLlibAgent(AgentSessionABC):
            self._current_result = self._agent.train()
            self._save_checkpoint()
        self._agent.stop()
+
        super().learn()
-        self.save(self.learning_path / f"rllib_{self._training_config.agent_identifier}_{self.timestamp_str}.zip")
+
+        # save agent
+        self.save()

    def evaluate(
        self,