fix(sklearn): probas_pred got removed since 1.7 (#879)

Author: shcheklein

pyproject.toml

@@ -44,7 +44,7 @@ dependencies = [
 
 [project.optional-dependencies]
 image = ["numpy", "pillow"]
-sklearn = ["scikit-learn"]
+sklearn = ["scikit-learn>=1.5.0"]
 plots = ["scikit-learn", "pandas", "numpy"]
 markdown = ["matplotlib"]
 tests = [

src/dvclive/plots/sklearn.py

@@ -54,7 +54,7 @@ def dump(self, val, **kwargs) -> None:
         from sklearn import metrics
 
         precision, recall, prc_thresholds = metrics.precision_recall_curve(
-            y_true=val[0], probas_pred=val[1], **kwargs
+            y_true=val[0], y_score=val[1], **kwargs
         )
 
         prc = {

tests/frameworks/test_huggingface.py

@@ -67,7 +67,7 @@ def __init__(self, a=0, b=0, double_output=False, random_torch=True, **kwargs):
 
 
 class RegressionPreTrainedModel(PreTrainedModel):
-    config_class = RegressionModelConfig
+    config_class = RegressionModelConfig  # type: ignore[assignment]
     base_model_prefix = "regression"
 
     def __init__(self, config):

tests/plots/test_sklearn.py

@@ -78,7 +78,7 @@ def test_log_prc_curve(tmp_dir, y_true_y_pred_y_score, mocker):
 
     live.log_sklearn_plot("precision_recall", y_true, y_score)
 
-    spy.assert_called_once_with(y_true=y_true, probas_pred=y_score)
+    spy.assert_called_once_with(y_true=y_true, y_score=y_score)
     assert (out / "precision_recall.json").exists()