Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Test dask demos. (#8557)

Browse Source 
Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan
2022-12-13 18:37:31 +08:00
committed by GitHub
parent 15a88ceef0
commit 40343c8ee1
4 changed files with 67 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/python-gpu/test_gpu_demos.py
View File

@@ -27,13 +27,3 @@ 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())
@pytest.mark.mgpu
def test_dask_training():
script = os.path.join(tm.demo_dir(__file__), 'dask', 'gpu_training.py')
cmd = ['python', script]
subprocess.check_call(cmd)