Skip to content

Commit

Permalink
TEST: add sklearn check_estimator test for MapieRegressor (#600)
Browse files Browse the repository at this point in the history
  • Loading branch information
Valentin-Laurent authored Jan 15, 2025
1 parent 0f511e6 commit 3d61d5a
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 0 deletions.
1 change: 1 addition & 0 deletions HISTORY.rst
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ History
* Documentation test: add a doc testing step (in MAKEFILE and CI)
* Increase max line length from 79 to 88 characters
* Bump wheel version
* Other minor evolutions

0.9.1 (2024-09-13)
------------------
Expand Down
6 changes: 6 additions & 0 deletions mapie/tests/test_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
)
from sklearn.pipeline import Pipeline, make_pipeline
from sklearn.preprocessing import OneHotEncoder
from sklearn.utils.estimator_checks import check_estimator
from sklearn.utils.validation import check_is_fitted
from typing_extensions import TypedDict

Expand Down Expand Up @@ -1039,3 +1040,8 @@ def test_deprecated_ensemble_regressor_fit_warning() -> None:
match=r".WARNING: EnsembleRegressor.fit is deprecated.*"
):
ens_reg.fit(X, y)


def test_mapie_regressor_sklearn_estim() -> None:
"""Test that MapieRegressor is an sklearn estimator"""
check_estimator(MapieRegressor())

0 comments on commit 3d61d5a

Please sign in to comment.