Tweak Status.exception() protocol

Make it return Optional[BaseException] like Future.exception()
coretl · Jun 9, 2023 · 2d7fdef · 2d7fdef
1 parent e38b2da
commit 2d7fdef
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/bluesky/protocols.py b/bluesky/protocols.py
@@ -1,5 +1,4 @@
 from abc import abstractmethod
-from asyncio import CancelledError
 from typing import (
     Any,
     Awaitable,
@@ -131,8 +130,6 @@ class Datum(TypedDict):
 T = TypeVar("T")
 SyncOrAsync = Union[T, Awaitable[T]]
 
-StatusException = Union[Exception, CancelledError]
-
 
 @runtime_checkable
 class Status(Protocol):
@@ -148,7 +145,7 @@ def add_callback(self, callback: Callable[["Status"], None]) -> None:
         ...
 
     @abstractmethod
-    def exception(self, timeout: Optional[float] = 0.0) -> Optional[StatusException]:
+    def exception(self, timeout: Optional[float] = 0.0) -> Optional[BaseException]:
         ...
 
     @property