From e1a5534a0d0dde1970f63b13f7af5edb93609798 Mon Sep 17 00:00:00 2001
From: Jev Kuznetsov <jev.kuznetsov@gmail.com>
Date: Tue, 7 May 2024 12:40:08 +0000
Subject: [PATCH] Refactor WebSocket client storage in MockRobot class

---
 examples/jupyter_ui/data_generator.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/jupyter_ui/data_generator.py b/examples/jupyter_ui/data_generator.py
index eb5ab6e..f0c3f3e 100644
--- a/examples/jupyter_ui/data_generator.py
+++ b/examples/jupyter_ui/data_generator.py
@@ -63,7 +63,8 @@ async def send_data(self) -> None:
                     "theta": round(self.theta, 3),
                 }
                 msg = json.dumps(data)
-                await asyncio.gather(*(client.send(msg) for client in self.clients))
+
+                websockets.broadcast(self.clients, msg)
                 log.info(f"Sending data: {msg}")
             await asyncio.sleep(1.0 / UPDATE_RATE)