Skip to content

Commit c4eb1f5

Browse files
perf: Move from enumerate to incremental
1 parent d0e8877 commit c4eb1f5

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/thread/thread.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -392,10 +392,13 @@ def wrapper(
392392
**kwargs: _Target_P.kwargs,
393393
) -> List[_Target_T]:
394394
computed: List[Data_Out] = []
395-
for i, data_entry in enumerate(data_chunk):
395+
396+
i = 0
397+
for data_entry in data_chunk:
396398
v = function(data_entry, *args, **kwargs)
397399
computed.append(v)
398400
self._threads[index].progress = round((i + 1) / len(data_chunk), 5)
401+
i += 1
399402

400403
self._completed += 1
401404
if self._completed == len(self._threads):
@@ -507,7 +510,7 @@ def start(self) -> None:
507510
i: v for i, v in self.overflow_kwargs.items() if i != 'name' and i != 'args'
508511
}
509512

510-
for i, data_chunk in enumerate(chunk_split(self.dataset, max_threads)):
513+
i = 0
511514
chunk_thread = Thread(
512515
target=self.function,
513516
args=[i, data_chunk, *parsed_args, *self.overflow_args],
@@ -516,6 +519,7 @@ def start(self) -> None:
516519
)
517520
self._threads.append(_ThreadWorker(chunk_thread, 0))
518521
chunk_thread.start()
522+
i += 1
519523

520524

521525
# Handle abrupt exit

0 commit comments

Comments
 (0)