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

15
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
generator = parallel_map(
func=func,
input_values=input_values,
chunk_size=chunk_size,
concurrency=concurrency,
)
return list( return list(
tqdm( tqdm(
parallel_map( generator,
func=func,
input_values=input_values,
chunk_size=chunk_size,
concurrency=concurrency,
),
total=len(input_values), total=len(input_values),
dynamic_ncols=True,
) )
) )