Merge pull request #32 from jdb78/maintenance/fix-tests

Make sklearn 0.24 compatiblen - try 2
jdb78 · Jan 2, 2021 · 6fa00e1 · 6fa00e1
2 parents 70bf946 + 4d1e17e
commit 6fa00e1
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/lipo/hyperparameter.py b/lipo/hyperparameter.py
@@ -120,10 +120,7 @@ def _run_search(self, evaluate_candidates):
             maximize=True,
             random_state=self.random_state,
         )
-        if self.multimetric_:
-            metric = self.refit
-        else:
-            metric = "score"
+        metric = "score"
 
         iteration = 0
         t = tqdm(total=self.n_iter, desc="Best params at iteration --- with score ----", mininterval=10.0)