From c85cdc91ca7b844f0914db2531e7488b30b25c8a Mon Sep 17 00:00:00 2001 From: Matthieu Jolimaitre Date: Thu, 7 Nov 2024 11:06:45 +0100 Subject: [PATCH] use threadpool as executor --- src/okipy/strategies.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/okipy/strategies.py b/src/okipy/strategies.py index 796389c..5d48251 100644 --- a/src/okipy/strategies.py +++ b/src/okipy/strategies.py @@ -1,6 +1,6 @@ from dataclasses import dataclass +from multiprocessing.pool import ThreadPool from typing import Callable, TypeVar -from multiprocessing import Pool T = TypeVar("T") @@ -17,7 +17,7 @@ class RunStrategy: class Parallel(RunStrategy): procs: None | int = None def run_all(self, items: list[T], oper: Callable[[T], O]) -> list[O]: - return Pool(self.procs).map(oper, items) + return ThreadPool(self.procs).map(oper, items) class Sequential(RunStrategy):