Skip to content

Commit 83a078b

Browse files
trivialfisrongou
andauthored
[backport] Fix sklearn test that calls a removed field (dmlc#8579) (dmlc#8636)
Co-authored-by: Rong Ou <[email protected]>
1 parent 575fba6 commit 83a078b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/python/test_with_sklearn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1030,7 +1030,7 @@ def test_pandas_input():
10301030
clf_isotonic = CalibratedClassifierCV(model, cv="prefit", method="isotonic")
10311031
clf_isotonic.fit(train, target)
10321032
assert isinstance(
1033-
clf_isotonic.calibrated_classifiers_[0].base_estimator, xgb.XGBClassifier
1033+
clf_isotonic.calibrated_classifiers_[0].estimator, xgb.XGBClassifier
10341034
)
10351035
np.testing.assert_allclose(np.array(clf_isotonic.classes_), np.array([0, 1]))
10361036

0 commit comments

Comments
 (0)