Fixed pytype error.
diff --git a/compiler_opt/distributed/worker.py b/compiler_opt/distributed/worker.py index 8fee04d..5e28aea 100644 --- a/compiler_opt/distributed/worker.py +++ b/compiler_opt/distributed/worker.py
@@ -64,7 +64,7 @@ return e -def lift_futures_through_list(future_list: WorkerFuture[List], +def lift_futures_through_list(future_list: WorkerFuture, expected_size: int) -> List[WorkerFuture]: """Convert Future[List] to List[Future].""" flattened = [concurrent.futures.Future() for _ in range(expected_size)]