This commit is contained in:
Anthony Sottile 2021-12-27 18:32:56 -05:00
parent d7ac975454
commit 83675fe768
2 changed files with 6 additions and 3 deletions

View file

@ -159,7 +159,8 @@ def xargs(
)
threads = min(len(partitions), target_concurrency)
with _thread_mapper(threads) as thread_map:
# https://github.com/python/mypy/issues/11852
with _thread_mapper(threads) as thread_map: # type: ignore
results = thread_map(run_cmd_partition, partitions)
for proc_retcode, proc_out, _ in results:

View file

@ -166,13 +166,15 @@ def test_xargs_concurrency():
def test_thread_mapper_concurrency_uses_threadpoolexecutor_map():
with xargs._thread_mapper(10) as thread_map:
# https://github.com/python/mypy/issues/11852
with xargs._thread_mapper(10) as thread_map: # type: ignore
_self = thread_map.__self__ # type: ignore
assert isinstance(_self, concurrent.futures.ThreadPoolExecutor)
def test_thread_mapper_concurrency_uses_regular_map():
with xargs._thread_mapper(1) as thread_map:
# https://github.com/python/mypy/issues/11852
with xargs._thread_mapper(1) as thread_map: # type: ignore
assert thread_map is map