8000 TST Add one more check · scikit-learn/scikit-learn@c8a0325 · GitHub
[go: up one dir, main page]

Skip to content

Commit c8a0325

Browse files
committed
TST Add one more check
1 parent 7f84de7 commit c8a0325

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

sklearn/ensemble/_hist_gradient_boosting/tests/test_gradient_boosting.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1684,6 +1684,8 @@ def test_categorical_different_order_same_model(dataframe_lib):
16841684
for predictor_1, predictor_2 in zip(hist_a_b._predictors, hist_b_a._predictors):
16851685
assert len(predictor_1[0].nodes) == len(predictor_2[0].nodes)
16861686

1687+
assert_allclose(hist_a_b.predict(df_a_b), hist_b_a.predict(df_b_a))
1688+
16871689

16881690
# TODO(1.6): Remove warning and change default in 1.6
16891691
def test_categorical_features_warn():

0 commit comments

Comments
 (0)
0