Update test_websockets.py

ag2ai · Dec 30, 2024 · d86ee94 · d86ee94
1 parent 22bbb85
commit d86ee94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/io/test_websockets.py b/test/io/test_websockets.py
@@ -172,7 +172,7 @@ def on_connect(iostream: IOWebsockets, success_dict: dict[str, bool] = success_d
                             print()
                             print(" - Received TERMINATE message.", flush=True)
                     except ConnectionClosed as e:
-                        print("Connection closed:", e)
+                        print("Connection closed:", e, flush=True)
                         break
 
         assert success_dict["success"]