Close pool as final step

runeharlyk · Jan 8, 2024 · 59ce707 · 59ce707
1 parent bc681a6
commit 59ce707
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/train_DQLAgent.py b/train_DQLAgent.py
@@ -61,6 +61,8 @@ def run_episodes(self, max_episode, max_steps):
             print(error)
             self.exit_program = True
         finally:
+            pool.close()
+            pool.join()
             return []
 
     def run_episode(self, episode, max_steps, pool):