diff --git a/memoize/statuses.py b/memoize/statuses.py index 5ebb17a..a6f3913 100644 --- a/memoize/statuses.py +++ b/memoize/statuses.py @@ -57,7 +57,7 @@ def mark_update_aborted(self, key: CacheKey, exception: Exception) -> None: update = self._updates_in_progress.pop(key) update.set_result(exception) - def await_updated(self, key: CacheKey) -> Awaitable[Union[CacheEntry | Exception]]: + def await_updated(self, key: CacheKey) -> Awaitable[Union[CacheEntry, Exception]]: """Waits (asynchronously) until update in progress has benn finished. Returns awaitable with the updated entry (or awaitable with an exception if update failed/timed-out).