#1386: fix bug with agent zip file not being saved after run

This commit is contained in:
Czar Echavez
2023-07-04 16:30:31 +01:00
parent c7de7bf21b
commit 818d64f330
3 changed files with 13 additions and 11 deletions

View File

@@ -142,6 +142,7 @@ class RLlibAgent(AgentSessionABC):
self._save_checkpoint()
self._agent.stop()
super().learn()
self.save(self.learning_path / f"rllib_{self._training_config.agent_identifier}_{self.timestamp_str}.zip")
def evaluate(
self,
@@ -162,10 +163,6 @@ class RLlibAgent(AgentSessionABC):
"""Load an agent from file."""
raise NotImplementedError
def save(self):
"""Save the agent."""
raise NotImplementedError
def export(self):
"""Export the agent to transportable file format."""
raise NotImplementedError