FIX: cast wrong conformity score to test failure

scikit-learn-contrib · May 5, 2023 · 23fedab · 23fedab
1 parent 4f088e5
commit 23fedab
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mapie/tests/test_classification.py b/mapie/tests/test_classification.py
@@ -900,7 +900,7 @@ def test_valid_cv(cv: Any) -> None:
     """Test that valid cv raises no errors."""
     model = LogisticRegression(multi_class="multinomial")
     model.fit(X_toy, y_toy)
-    mapie_clf = MapieClassifier(estimator=model, cv=cv, random_state=42)
+    mapie_clf = MapieClassifier(estimator=model, cv=cv)
     mapie_clf.fit(X_toy, y_toy)
     mapie_clf.predict(X_toy, alpha=0.5)
 

diff --git a/mapie/tests/test_regression.py b/mapie/tests/test_regression.py
@@ -618,7 +618,8 @@ def test_wrong_not_fitted_conformity_score(cv: Any) -> None:
 
 
 def test_wrong_conformity_score() -> None:
-    mapie_reg = MapieRegressor(conformity_score=object())
+    wrong_conformity_score = cast(ConformityScore, object())
+    mapie_reg = MapieRegressor(conformity_score=wrong_conformity_score)
     with pytest.raises(
         ValueError,
         match=r".*Invalid conformity_score argument.*",