diff --git a/tests/worker/test_reworker.py b/tests/worker/test_reworker.py index 519fa91c9..0f2b3be7b 100644 --- a/tests/worker/test_reworker.py +++ b/tests/worker/test_reworker.py @@ -380,7 +380,7 @@ def assert_running_count_plan_produces_ordered_worker_and_data_events( def take_n_events( stream: EventStream[E, Any], num: int, -) -> "Future[list[E]]": +) -> Future[list[E]]: count = itertools.count() return take_events(stream, lambda _: next(count) >= num) @@ -388,7 +388,7 @@ def take_n_events( def take_events( stream: EventStream[E, Any], cutoff_predicate: Callable[[E], bool], -) -> "Future[list[E]]": +) -> Future[list[E]]: events: list[E] = [] future: Future[list[E]] = Future() @@ -405,7 +405,7 @@ def on_event(event: E, event_id: str | None) -> None: def take_events_from_streams( streams: list[EventStream[Any, int]], cutoff_predicate: Callable[[Any], bool], -) -> "Future[list[Any]]": +) -> Future[list[Any]]: """Returns a collated list of futures for events in numerous event streams. The support for generic and algebraic types doesn't appear to extend to