Apply suggestions from code review

2023-11-17 20:30:07 +00:00
parent c5b4ae45be
commit 9d0a98b221
3 changed files with 7 additions and 5 deletions
--- a/src/primaite/game/policy/sb3.py
+++ b/src/primaite/game/policy/sb3.py
@@ -74,10 +74,6 @@ class SB3Policy(PolicyABC, identifier="SB3"):
        """Load agent from a checkpoint."""
        self._agent = self._agent_class.load(model_path, env=self.session.env)

-    def close(self) -> None:
-        """Close the agent."""
-        pass
-
    @classmethod
    def from_config(cls, config: "TrainingOptions", session: "PrimaiteSession") -> "SB3Policy":
        """Create an agent from config file."""