From fd0a474cd1f6bc1d678b706d8384eb0cd53e64e8 Mon Sep 17 00:00:00 2001 From: hvrai Date: Tue, 4 Jun 2024 08:20:12 +0530 Subject: [PATCH] Fix memory leak in websocket handler --- ads/aqua/extension/ui_websocket_handler.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ads/aqua/extension/ui_websocket_handler.py b/ads/aqua/extension/ui_websocket_handler.py index 701a9b023..fa3a4e254 100644 --- a/ads/aqua/extension/ui_websocket_handler.py +++ b/ads/aqua/extension/ui_websocket_handler.py @@ -112,6 +112,7 @@ def on_message_processed(self, future: concurrent.futures.Future): ) raise e finally: + self.future_message_map.pop(future) # Send the response back to the client on the event thread IOLoop.current().run_sync(lambda: self.write_message(response.to_json()))