Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-08 02:34:06 -04:00
Code Issues Packages Projects Releases Wiki Activity

Handle empty iterables

Browse Source
This commit is contained in:
Kovid Goyal 2025-04-06 14:05:32 +05:30
parent 85b107b128
commit 5d4f56d4fa
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/calibre/utils/forked_map.py
View File

@ -117,6 +117,8 @@ def forked_map(fn: Callable[[T], R], iterable: T, *iterables: T, timeout: int |
system libraries. system libraries.
''' '''
num_items = len(iterable) + sum(map(len, iterables)) num_items = len(iterable) + sum(map(len, iterables))
if num_items < 1:
return
if num_workers <= 0: if num_workers <= 0:
num_workers = max(1, min(num_items, os.cpu_count())) num_workers = max(1, min(num_items, os.cpu_count()))
chunk_size = max(1, num_items // num_workers) chunk_size = max(1, num_items // num_workers)
@ -151,11 +153,11 @@ def forked_map(fn: Callable[[T], R], iterable: T, *iterables: T, timeout: int |
pos += 1 pos += 1
else: else:
cache[result.id] = result cache[result.id] = result
if count < num_items:
raise OSError(f'Forked workers exited producing only {count} out of {num_items} results')
while r := cache.pop(pos, None): while r := cache.pop(pos, None):
yield r.value yield r.value
pos += 1 pos += 1
if pos < num_items:
raise OSError(f'Forked workers exited producing only {pos} out of {len(iterables)} results')
forked_map_is_supported = hasattr(os, 'fork') forked_map_is_supported = hasattr(os, 'fork')