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

Disable pytest-timeout for now. (#8348)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-10-17 23:06:10 +08:00
committed by GitHub
parent fcddbc9264
commit 2176e511fc
13 changed files with 107 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/python/test_data_iterator.py
View File

@@ -1,12 +1,15 @@
import xgboost as xgb
from xgboost.data import SingleBatchInternalIter as SingleBatch
import numpy as np
from testing import IteratorForTest, non_increasing, make_batches
import pytest
from hypothesis import given, strategies, settings
from hypothesis import given, settings, strategies
from scipy.sparse import csr_matrix
from testing import IteratorForTest, make_batches, non_increasing
from xgboost.data import SingleBatchInternalIter as SingleBatch
import xgboost as xgb
from xgboost import testing
pytestmark = testing.timeout(30)
pytestmark = pytest.mark.timeout(30)
def test_single_batch(tree_method: str = "approx") -> None:
from sklearn.datasets import load_breast_cancer