Demo for experimental categorical data support. (#7213)

2021-09-15 08:20:12 +08:00
parent 3515931305
commit d997c967d5
2 changed files with 81 additions and 0 deletions
--- a/tests/python-gpu/test_gpu_demos.py
+++ b/tests/python-gpu/test_gpu_demos.py
@@ -20,6 +20,12 @@ def test_update_process_demo():
    subprocess.check_call(cmd)


+def test_categorical_demo():
+    script = os.path.join(td.PYTHON_DEMO_DIR, 'categorical.py')
+    cmd = ['python', script]
+    subprocess.check_call(cmd)
+
+
@pytest.mark.skipif(**tm.no_dask())
@pytest.mark.skipif(**tm.no_dask_cuda())
@pytest.mark.skipif(**tm.no_cupy())