Skip to content

Commit dbc092e

Browse files
committed
add inner_max_num_threads=1
1 parent 55494bf commit dbc092e

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

monte-cover/src/montecover/base.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
import numpy as np
1313
import pandas as pd
1414
import yaml
15-
from joblib import Parallel, delayed
15+
from joblib import Parallel, delayed, parallel_backend
1616

1717

1818
class BaseSimulation(ABC):
@@ -111,9 +111,10 @@ def run_simulation(self, n_jobs=None):
111111

112112
else:
113113
self.logger.info(f"Starting parallel execution with n_jobs={n_jobs}")
114-
results = Parallel(n_jobs=n_jobs, verbose=10)(
115-
delayed(self._process_repetition)(i_rep) for i_rep in range(self.repetitions) if not self._stop_simulation()
116-
)
114+
with parallel_backend("loky", inner_max_num_threads=1):
115+
results = Parallel(n_jobs=n_jobs, verbose=10)(
116+
delayed(self._process_repetition)(i_rep) for i_rep in range(self.repetitions) if not self._stop_simulation()
117+
)
117118

118119
# Process results from parallel execution
119120
for worker_results in results:

0 commit comments

Comments
 (0)