[pyspark] Use quantile dmatrix. (#8284)

2022-10-12 20:38:53 +08:00
parent ce0382dcb0
commit 97a5b088a5
9 changed files with 225 additions and 120 deletions
--- a/tests/python-gpu/test_gpu_spark/test_data.py
+++ b/tests/python-gpu/test_gpu_spark/test_data.py
@@ -20,4 +20,6 @@ from test_spark.test_data import run_dmatrix_ctor

@pytest.mark.skipif(**tm.no_cudf())
 def test_qdm_ctor() -> None:
-    run_dmatrix_ctor(True)
+    run_dmatrix_ctor(is_dqm=True, on_gpu=True)
+    with pytest.raises(AssertionError):
+        run_dmatrix_ctor(is_dqm=False, on_gpu=True)