Make devices connect with a timeout

DiamondLightSource · Oct 25, 2023 · 22e2914 · 22e2914
1 parent d80eca0
commit 22e2914
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/blueapi/core/context.py b/src/blueapi/core/context.py
@@ -1,3 +1,4 @@
+import asyncio
 import functools
 import logging
 from dataclasses import dataclass, field
@@ -114,7 +115,7 @@ async def connect_devices(self, sim: bool = False) -> None:
                 coros[device_name] = device.connect(sim)
 
         if len(coros) > 0:
-            await wait_for_connection(**coros)
+            await asyncio.wait(wait_for_connection(**coros), timeout=30.0)
 
     def with_plan_module(self, module: ModuleType) -> None:
         """