UTMIST · kseto06 · Oct 25, 2025 · Oct 25, 2025 · Oct 30, 2025
diff --git a/environment/__pycache__/WarehouseBrawl.cpython-313.pyc b/environment/__pycache__/WarehouseBrawl.cpython-313.pyc
diff --git a/environment/__pycache__/agent.cpython-313.pyc b/environment/__pycache__/agent.cpython-313.pyc
diff --git a/environment/__pycache__/environment.cpython-312.pyc b/environment/__pycache__/environment.cpython-312.pyc
diff --git a/environment/__pycache__/environment.cpython-313.pyc b/environment/__pycache__/environment.cpython-313.pyc
diff --git a/environment/agent.py b/environment/agent.py
@@ -1026,7 +1026,10 @@ def train(agent: Agent,
         agent.learn(env, total_timesteps=train_timesteps, verbose=1)
         base_env.on_training_end()
     except KeyboardInterrupt:
-        pass
+        if save_handler is not None:
+            save_handler.agent.update_num_timesteps(save_handler.num_timesteps)
+            save_handler.save_agent()
+
 
     env.close()
 

diff --git a/user/pvp_match.py b/user/pvp_match.py