From eed8f8cdfd724443accaa070325f8bf025086305 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 24 Mar 2022 19:23:29 +0200
Subject: [PATCH] Revert Optuna deprecation fix (#613)

---
 clearml/automation/optuna/optuna.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/clearml/automation/optuna/optuna.py b/clearml/automation/optuna/optuna.py
index 28c6b665..a69b978f 100644
--- a/clearml/automation/optuna/optuna.py
+++ b/clearml/automation/optuna/optuna.py
@@ -1,4 +1,3 @@
-from concurrent.futures import ThreadPoolExecutor
 from time import sleep
 from typing import Any, Optional, Sequence
 
@@ -175,9 +174,8 @@ class OptimizerOptuna(SearchStrategy):
             sleep_interval=int(self.pool_period_minutes * 60),
             config_space=config_space,
         )
-        with ThreadPoolExecutor(max_workers=self._num_concurrent_workers) as executor:
-            for _ in range(self._num_concurrent_workers):
-                executor.submit(self._study.optimize, self._objective.objective, n_trials=self.total_max_jobs)
+        self._study.optimize(
+            self._objective.objective, n_trials=self.total_max_jobs, n_jobs=self._num_concurrent_workers)
 
     def stop(self):
         # type: () -> ()