Refactor WebSocket client storage in MockRobot class

rox-automation · May 7, 2024 · e1a5534 · e1a5534
1 parent cd16789
commit e1a5534
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)