From 42e6fbb0dbe1f2b6e9dede151ff1e5a810584f84 Mon Sep 17 00:00:00 2001 From: Rong Ou Date: Fri, 9 Dec 2022 13:06:44 -0800 Subject: [PATCH] Fix sklearn test that calls a removed field (#8579) --- tests/python/test_with_sklearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/test_with_sklearn.py b/tests/python/test_with_sklearn.py index ff353400b..2fa20e284 100644 --- a/tests/python/test_with_sklearn.py +++ b/tests/python/test_with_sklearn.py @@ -1025,7 +1025,7 @@ def test_pandas_input(): clf_isotonic = CalibratedClassifierCV(model, cv="prefit", method="isotonic") clf_isotonic.fit(train, target) assert isinstance( - clf_isotonic.calibrated_classifiers_[0].base_estimator, xgb.XGBClassifier + clf_isotonic.calibrated_classifiers_[0].estimator, xgb.XGBClassifier ) np.testing.assert_allclose(np.array(clf_isotonic.classes_), np.array([0, 1]))