fix: wait overload

phi-friday · Sep 26, 2024 · b0f3d8d · b0f3d8d
1 parent 7921909
commit b0f3d8d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/timeout_executor/result.py b/src/timeout_executor/result.py
@@ -74,13 +74,15 @@ def has_result(self) -> bool:
         """check if result is available"""
         return self._result is not SENTINEL
 
+    @overload
+    def wait(self, timeout: float | None = None) -> Awaitable[None]: ...
     @overload
     def wait(
-        self, timeout: float | None = None, *, do_async: Literal[True] = ...
+        self, timeout: float | None = None, *, do_async: Literal[True]
     ) -> Awaitable[None]: ...
     @overload
     def wait(
-        self, timeout: float | None = None, *, do_async: Literal[False] = ...
+        self, timeout: float | None = None, *, do_async: Literal[False]
     ) -> None: ...
     @overload
     def wait(