|
@@ -2,7 +2,7 @@ import atexit
|
|
|
from concurrent.futures import Executor, ThreadPoolExecutor
|
|
|
import concurrent
|
|
|
import concurrent.futures
|
|
|
-
|
|
|
+from concurrent.futures import as_completed
|
|
|
from typing import List
|
|
|
|
|
|
|
|
@@ -53,7 +53,7 @@ def run_with_executor(executor: Executor, func, tasks: List, timeout: int):
|
|
|
results.append(future.result())
|
|
|
return results
|
|
|
"""
|
|
|
-
|
|
|
+ """
|
|
|
results = []
|
|
|
# Iterate over tasks and execute them sequentially
|
|
|
for task in tasks:
|
|
@@ -71,12 +71,11 @@ def run_with_executor(executor: Executor, func, tasks: List, timeout: int):
|
|
|
futures = [executor.submit(lambda args: func(*args), task) for task in tasks]
|
|
|
|
|
|
results = []
|
|
|
- for future in concurrent.futures.as_completed(futures, timeout=timeout):
|
|
|
- if future.exception():
|
|
|
- raise future.exception()
|
|
|
-
|
|
|
- if future.done():
|
|
|
- results.append(future.result())
|
|
|
+ for future in as_completed(futures, timeout=timeout):
|
|
|
+ try:
|
|
|
+ result = future.result(timeout=timeout)
|
|
|
+ results.append(result)
|
|
|
+ except Exception as e:
|
|
|
+ print(e)
|
|
|
|
|
|
return results
|
|
|
- """
|