xgboost/tests/python-gpu/test_gpu_demos.py
Jiaming Yuan a3ec964346
Accept iterator in device dmatrix. (#5783)
* Remove Device DMatrix.
2020-07-07 21:44:48 +08:00

12 lines
264 B
Python

import os
import subprocess
import sys
sys.path.append("tests/python")
import test_demos as td # noqa
def test_data_iterator():
script = os.path.join(td.PYTHON_DEMO_DIR, 'data_iterator.py')
cmd = ['python', script]
subprocess.check_call(cmd)