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

Import data lazily during tests. (#9176)

Browse Source
This commit is contained in:
Jiaming Yuan
2023-05-23 03:58:31 +08:00
committed by GitHub
parent 6274fba0a5
commit 3913ff470f
8 changed files with 66 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py
View File

@@ -243,7 +243,7 @@ class TestDistributedGPU:
@given(
params=hist_parameter_strategy,
num_rounds=strategies.integers(1, 20),
dataset=tm.dataset_strategy,
dataset=tm.make_dataset_strategy(),
dmatrix_type=strategies.sampled_from(
[dxgb.DaskDMatrix, dxgb.DaskQuantileDMatrix]
),

9
tests/test_distributed/test_with_dask/test_with_dask.py
View File

@@ -1458,9 +1458,10 @@ class TestWithDask:
else:
assert history[-1] < history[0]
@given(params=hist_parameter_strategy,
dataset=tm.dataset_strategy)
@settings(deadline=None, max_examples=10, suppress_health_check=suppress, print_blob=True)
@given(params=hist_parameter_strategy, dataset=tm.make_dataset_strategy())
@settings(
deadline=None, max_examples=10, suppress_health_check=suppress, print_blob=True
)
def test_hist(
self, params: Dict, dataset: tm.TestDataset, client: "Client"
) -> None:
@@ -1524,7 +1525,7 @@ class TestWithDask:
rmse = result["history"]["Valid"]["rmse"][-1]
assert rmse < 32.0
@given(params=hist_parameter_strategy, dataset=tm.dataset_strategy)
@given(params=hist_parameter_strategy, dataset=tm.make_dataset_strategy())
@settings(
deadline=None, max_examples=10, suppress_health_check=suppress, print_blob=True
)