Import data lazily during tests. (#9176)

2023-05-23 03:58:31 +08:00
parent 6274fba0a5
commit 3913ff470f
8 changed files with 66 additions and 51 deletions
--- a/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py
+++ b/tests/test_distributed/test_gpu_with_dask/test_gpu_with_dask.py
@@ -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]
        ),