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
|
"""Copyright 2024, XGBoost contributors"""
import dask
import pytest
from distributed import Client
from xgboost import testing as tm
from xgboost.testing import dask as dtm
pytestmark = [
pytest.mark.skipif(**tm.no_dask()),
pytest.mark.skipif(**tm.no_dask_cuda()),
tm.timeout(120),
]
@pytest.mark.filterwarnings("error")
def test_no_group_split(local_cuda_client: Client) -> None:
with dask.config.set(
{
"array.backend": "cupy",
"dataframe.backend": "cudf",
}
):
dtm.check_no_group_split(local_cuda_client, "cuda")
|
