Fix scorer representation test for newer version of scikit-learn

Author: AnesBenmerzoug

tests/utils/test_score.py

@@ -1,5 +1,7 @@
 import numpy as np
+import sklearn
 from numpy.typing import NDArray
+from packaging import version
 
 from pydvl.utils.score import Scorer, compose_score, squashed_r2, squashed_variance
 
@@ -24,7 +26,13 @@ def test_scorer():
     """Tests the Scorer class."""
     scorer = Scorer("r2")
     assert str(scorer) == "r2"
-    assert repr(scorer) == "R2 (scorer=make_scorer(r2_score))"
+    if version.parse(sklearn.__version__) >= version.parse("1.4.0"):
+        assert (
+            repr(scorer)
+            == "R2 (scorer=make_scorer(r2_score, response_method='predict'))"
+        )
+    else:
+        assert repr(scorer) == "R2 (scorer=make_scorer(r2_score))"
 
     coef = np.array([1, 2])
     X = np.array([[1, 2], [3, 4]])

tests/value/shapley/test_classwise.py

@@ -3,7 +3,9 @@
 import numpy as np
 import pandas as pd
 import pytest
+import sklearn
 from numpy.typing import NDArray
+from packaging import version
 
 from pydvl.utils import Dataset, Utility, powerset
 from pydvl.value import MaxChecks, ValuationResult
@@ -165,7 +167,13 @@ def test_classwise_scorer_representation():
 
     scorer = ClasswiseScorer("accuracy", initial_label=0)
     assert str(scorer) == "classwise accuracy"
-    assert repr(scorer) == "ClasswiseAccuracy (scorer=make_scorer(accuracy_score))"
+    if version.parse(sklearn.__version__) >= version.parse("1.4.0"):
+        assert (
+            repr(scorer)
+            == "ClasswiseAccuracy (scorer=make_scorer(accuracy_score, response_method='predict'))"
+        )
+    else:
+        assert repr(scorer) == "ClasswiseAccuracy (scorer=make_scorer(accuracy_score))"
 
 
 @pytest.mark.parametrize("n_element, left_margin, right_margin", [(101, 0.3, 0.4)])