From 8375fd8ae05219b1e9e0f6d15b53272346066b03 Mon Sep 17 00:00:00 2001
From: fheinrich <fheinrich@techfak.uni-bielefeld.de>
Date: Mon, 29 Jan 2024 13:40:37 +0100
Subject: [PATCH] Fixed exit of main and pygame gui

---
 overcooked_simulator/__main__.py                  | 2 +-
 overcooked_simulator/gui_2d_vis/overcooked_gui.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/overcooked_simulator/__main__.py b/overcooked_simulator/__main__.py
index f68c506d..f6c3ab9f 100644
--- a/overcooked_simulator/__main__.py
+++ b/overcooked_simulator/__main__.py
@@ -44,7 +44,7 @@ def main(cli_args=None):
             game_server.terminate()
         if pygame_gui is not None and pygame_gui.is_alive():
             print("Terminate pygame gui")
-            game_server.terminate()
+            pygame_gui.terminate()
         time.sleep(0.1)
 
 
diff --git a/overcooked_simulator/gui_2d_vis/overcooked_gui.py b/overcooked_simulator/gui_2d_vis/overcooked_gui.py
index 8e4e4025..1b757c65 100644
--- a/overcooked_simulator/gui_2d_vis/overcooked_gui.py
+++ b/overcooked_simulator/gui_2d_vis/overcooked_gui.py
@@ -669,6 +669,7 @@ class PyGameGUI:
             except (KeyboardInterrupt, SystemExit):
                 self.running = False
 
+        self.disconnect_websockets()
         pygame.quit()
         sys.exit()
 
-- 
GitLab