Skip to content

Commit

Permalink
Merge pull request #1510 from Avaiga/feature/#1470-setup-spawn-method…
Browse files Browse the repository at this point in the history
…-for-subprocess-creation

#1470 setup spawn method for creating subprocesses
  • Loading branch information
toan-quach authored Jul 15, 2024
2 parents 09ceb85 + 7f4989f commit 32e68a5
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
# an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
# specific language governing permissions and limitations under the License.

import multiprocessing as mp
from concurrent.futures import Executor, ProcessPoolExecutor
from functools import partial
from threading import Lock
Expand All @@ -33,9 +34,8 @@ def __init__(self, orchestrator: _AbstractOrchestrator, subproc_initializer: Opt
super().__init__(orchestrator)
max_workers = Config.job_config.max_nb_of_workers or self._DEFAULT_MAX_NB_OF_WORKERS
self._executor: Executor = ProcessPoolExecutor(
max_workers=max_workers,
initializer=subproc_initializer,
) # type: ignore
max_workers=max_workers, initializer=subproc_initializer, mp_context=mp.get_context("spawn")
)
self._nb_available_workers = self._executor._max_workers # type: ignore

def _can_execute(self) -> bool:
Expand Down

0 comments on commit 32e68a5

Please sign in to comment.