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

5
tests/python-gpu/test_gpu_linear.py
View File

@@ -28,8 +28,7 @@ def train_result(param, dmat, num_rounds):
class TestGPULinear:
@given(parameter_strategy, strategies.integers(10, 50),
tm.dataset_strategy)
@given(parameter_strategy, strategies.integers(10, 50), tm.make_dataset_strategy())
@settings(deadline=None, max_examples=20, print_blob=True)
def test_gpu_coordinate(self, param, num_rounds, dataset):
assume(len(dataset.y) > 0)
@@ -45,7 +44,7 @@ class TestGPULinear:
@given(
parameter_strategy,
strategies.integers(10, 50),
tm.dataset_strategy,
tm.make_dataset_strategy(),
strategies.floats(1e-5, 0.8),
strategies.floats(1e-5, 0.8)
)