Remove all use of DeviceQuantileDMatrix. (#8665)

2023-01-17 00:04:10 +08:00
parent 0ae8df9a65
commit d6018eb4b9
10 changed files with 57 additions and 53 deletions
--- a/tests/python/test_updaters.py
+++ b/tests/python/test_updaters.py
@@ -173,7 +173,7 @@ class TestTreeMethod:

            X, y = cp.array(X), cp.array(y)
            with pytest.raises(ValueError):
-                Xy = xgb.DeviceQuantileDMatrix(X, y, feature_types=["c"] * 10)
+                Xy = xgb.QuantileDMatrix(X, y, feature_types=["c"] * 10)

    def test_invalid_category(self) -> None:
        self.run_invalid_category("approx")