From 52e1d5605fcc2de9e2b1e331b30e3dda4b1492b7 Mon Sep 17 00:00:00 2001 From: Christian Schwarz Date: Mon, 8 Jan 2024 18:29:29 +0000 Subject: [PATCH] work_queue abstraction: typing fixes --- test_runner/fixtures/work_queue.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test_runner/fixtures/work_queue.py b/test_runner/fixtures/work_queue.py index 68c190d25d..4e844436d8 100644 --- a/test_runner/fixtures/work_queue.py +++ b/test_runner/fixtures/work_queue.py @@ -21,10 +21,10 @@ def do(nthreads: int, inputs: List[T], work_fn: Callable[[T], U]) -> List[U]: if item is None: return output = work_fn(item._item) - output_queue.put(Item(output)) + output_queue.put(output) input_queue: queue.Queue[Optional["Item[T]"]] = queue.Queue() - output_queue: queue.Queue[Optional["Item[U]"]] = queue.Queue() + output_queue: queue.Queue[U] = queue.Queue() for t in inputs: input_queue.put(Item(t)) workers = []