Improve typing of with_timeout

sdss · Dec 30, 2024 · 50d7c8a · 50d7c8a
1 parent 18812db
commit 50d7c8a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,12 @@
 # Changelog
 
+## Next version
+
+### ⚙️ Engineering
+
+* Improved typing of `with_timeout` function.
+
+
 ## 0.5.5 - December 27, 2024
 
 ### 🔧 Fixed

diff --git a/src/lvmopstools/utils.py b/src/lvmopstools/utils.py
@@ -107,7 +107,7 @@ def is_notebook() -> bool:
 
 
 async def with_timeout(
-    coro: Coroutine[Any, Any, T],
+    coro: asyncio.Future[T] | Coroutine[Any, Any, T],
     timeout: float | None,
     raise_on_timeout: bool = True,
 ) -> T | None: