From 1e7f5b62f3ca1feced1040ac83b9b2e994fd079c Mon Sep 17 00:00:00 2001
From: Chris McCarthy <chris.mccarthy@methods.co.uk>
Date: Thu, 6 Jul 2023 13:56:12 +0100
Subject: [PATCH] #1963 - Made RLlib and SB3 agents save at the end of each
 learning session by default using a common file naming format. Also now
 agents only checkpoint every n and not on the final episode

---
 src/primaite/agents/rllib.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/primaite/agents/rllib.py b/src/primaite/agents/rllib.py
index 32dc3dc0..ce072a03 100644
--- a/src/primaite/agents/rllib.py
+++ b/src/primaite/agents/rllib.py
@@ -186,6 +186,7 @@ class RLlibAgent(AgentSessionABC):
             checkpoint_dir = file
             break
 
+        # Zip the folder
         shutil.make_archive(
             str(self._saved_agent_path).replace(".zip", ""),
             "zip",