andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Refactor

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-13 08:33:22 +02:00
parent 0585a4e4f9
commit 6106816999
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
great_ai/utilities/parallel_map/simple_parallel_map.py
View file

@ -46,15 +46,16 @@ def simple_parallel_map(
""" """
input_values = list(input_values) # in case the input is mistakenly not a sequence input_values = list(input_values) # in case the input is mistakenly not a sequence
return list( generator = parallel_map(
tqdm(
parallel_map(
func=func, func=func,
input_values=input_values, input_values=input_values,
chunk_size=chunk_size, chunk_size=chunk_size,
concurrency=concurrency, concurrency=concurrency,
), )
return list(
tqdm(
generator,
total=len(input_values), total=len(input_values),
dynamic_ncols=True,
) )
) )