diff --git a/tests/python/test_demos.py b/tests/python/test_demos.py index 0faff4a73570..c4e4db4b5b7f 100644 --- a/tests/python/test_demos.py +++ b/tests/python/test_demos.py @@ -165,6 +165,15 @@ def test_multioutput_reg() -> None: subprocess.check_call(cmd) +@pytest.mark.skipif(**tm.no_sklearn()) +def test_multioutput_reduced_gradient_demo() -> None: + # Runs with the default `--device cpu`; `cupy` is only imported on the + # (unused here) `--device cuda` code paths. + script = os.path.join(PYTHON_DEMO_DIR, "multioutput_reduced_gradient.py") + cmd = [PYTHON, script] + subprocess.check_call(cmd) + + @pytest.mark.skipif(**tm.no_sklearn()) def test_prediction_intervals() -> None: script = os.path.join(PYTHON_DEMO_DIR, "prediction_intervals.py")