fix max threads to 20

andreatramacere · May 2, 2024 · c4de8ce · c4de8ce
1 parent 5c3e3f3
commit c4de8ce
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jetset/jet_model.py b/jetset/jet_model.py
@@ -105,8 +105,9 @@ def __init__(self,
         self.nu_grid_size=self._get_nu_grid_size_blob()
         N=multiprocessing.cpu_count()
         if N>2:
+            N=multiprocessing.cpu_count()
             N=min(N,20)
-        self.set_num_c_threads(N=multiprocessing.cpu_count())
+        self.set_num_c_threads(N=N)
         if jet_workplace is None:
             jet_workplace=WorkPlace()
             out_dir= jet_workplace.out_dir + '/' + self.name + '_jet_prod/'