From a4bf4d05230474ad14243d67ac7f8c938f670e58 Mon Sep 17 00:00:00 2001 From: Scott Date: Wed, 2 Feb 2022 09:14:12 -0800 Subject: More type annotations. --- smart_future.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) (limited to 'smart_future.py') diff --git a/smart_future.py b/smart_future.py index 604c149..1f6e6f0 100644 --- a/smart_future.py +++ b/smart_future.py @@ -5,7 +5,7 @@ import concurrent import concurrent.futures as fut import logging import traceback -from typing import Callable, List, TypeVar +from typing import Callable, List, Set, TypeVar from overrides import overrides @@ -27,11 +27,11 @@ def wait_any( ): real_futures = [] smart_future_by_real_future = {} - completed_futures = set() - for f in futures: - assert type(f) == SmartFuture - real_futures.append(f.wrapped_future) - smart_future_by_real_future[f.wrapped_future] = f + completed_futures: Set[fut.Future] = set() + for x in futures: + assert type(x) == SmartFuture + real_futures.append(x.wrapped_future) + smart_future_by_real_future[x.wrapped_future] = x while len(completed_futures) != len(real_futures): newly_completed_futures = concurrent.futures.as_completed(real_futures) @@ -59,9 +59,9 @@ def wait_all( log_exceptions: bool = True, ) -> None: real_futures = [] - for f in futures: - assert type(f) == SmartFuture - real_futures.append(f.wrapped_future) + for x in futures: + assert type(x) == SmartFuture + real_futures.append(x.wrapped_future) (done, not_done) = concurrent.futures.wait( real_futures, timeout=None, return_when=concurrent.futures.ALL_COMPLETED -- cgit v1.3