_MultimetricScorer was moved out of sklearn.model_selection.validatio…

…n._score so we have to do it
sustainable-processes · Oct 22, 2024 · 9db3516 · 9db3516
1 parent ecb8b88
commit 9db3516
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/summit/benchmarks/experimental_emulator.py b/summit/benchmarks/experimental_emulator.py
@@ -41,7 +41,7 @@
 )
 from sklearn.utils import check_array, _safe_indexing
 from sklearn.utils.parallel import delayed
-from sklearn.metrics._scorer import _check_multimetric_scoring
+from sklearn.metrics._scorer import _check_multimetric_scoring, _MultimetricScorer
 
 from scipy.sparse import issparse
 
@@ -366,6 +366,7 @@ def test(self, **kwargs):
                 scorers = check_scoring(predictor, scoring)
             else:
                 scorers = _check_multimetric_scoring(predictor, scoring)
+                scorers = _MultimetricScorer(scorers=scorers)
             scores_list.append(_score(predictor, X_test, y_test, scorers, score_params = None))
         scores_dict = _aggregate_score_dicts(scores_list)
         for name in scoring: