diff --git a/python-package/xgboost/testing/updater.py b/python-package/xgboost/testing/updater.py index 702148e06edc..b92d7a9fa7e3 100644 --- a/python-package/xgboost/testing/updater.py +++ b/python-package/xgboost/testing/updater.py @@ -228,7 +228,6 @@ def check_extmem_qdm( Xy = xgb.QuantileDMatrix(it) booster = xgb.train({"device": device}, Xy, num_boost_round=8) - # Get cuts from ellpack without CPU-GPU interpolation is not yet supported. cut_it = Xy_it.get_quantile_cut() cut = Xy.get_quantile_cut() np.testing.assert_allclose(cut_it[0], cut[0])