Align number of jobs and trials (#34)

Co-authored-by: Benedikt Volkel <[email protected]>
AliceO2Group · Sep 27, 2022 · 69d8784 · 69d8784
1 parent 851d6fa
commit 69d8784
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/o2tuner/optimise.py b/src/o2tuner/optimise.py
@@ -52,6 +52,10 @@ def optimise(objective, optuna_config, *, work_dir="o2tuner_optimise", user_conf
         run_serial = True
         jobs = 1
 
+    if trials < jobs:
+        LOG.info(f"Attempt to do {trials} trials, hence reducing the number of jobs from {jobs} to {trials}")
+        jobs = trials
+
     trials_list = floor(trials / jobs)
     trials_list = [trials_list] * jobs
     # add the left-over trials simply to the last job for now