1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
import os
import subprocess
import pytest
from xgboost import testing as tm
@pytest.mark.skipif(**tm.no_dask())
def test_dask_cpu_training_demo():
script = os.path.join(tm.demo_dir(__file__), "dask", "cpu_training.py")
cmd = ["python", script]
subprocess.check_call(cmd)
@pytest.mark.skipif(**tm.no_dask())
def test_dask_cpu_survival_demo():
script = os.path.join(tm.demo_dir(__file__), "dask", "cpu_survival.py")
cmd = ["python", script]
subprocess.check_call(cmd)
# Not actually run on CI due to missing dask_ml.
@pytest.mark.skipif(**tm.no_dask())
@pytest.mark.skipif(**tm.no_dask_ml())
def test_dask_callbacks_demo():
script = os.path.join(tm.demo_dir(__file__), "dask", "dask_callbacks.py")
cmd = ["python", script]
subprocess.check_call(cmd)
@pytest.mark.skipif(**tm.no_dask())
def test_dask_sklearn_demo():
script = os.path.join(tm.demo_dir(__file__), "dask", "sklearn_cpu_training.py")
cmd = ["python", script]
subprocess.check_call(cmd)
|
