#!/usr/bin/env python3
from __future__ import annotations
+
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
+import id_generator
+
# This module is commonly used by others in here and should avoid
# taking any unnecessary dependencies back on them.
from deferred_operand import DeferredOperand
-import id_generator
logger = logging.getLogger(__name__)
):
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)
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