Check inf in data for all types of DMatrix. (#8911)

This commit is contained in:
Jiaming Yuan
2023-03-15 11:24:35 +08:00
committed by GitHub
parent 72e8331eab
commit f186c87cf9
11 changed files with 118 additions and 45 deletions

View File

@@ -15,7 +15,7 @@ from xgboost.testing import (
make_sparse_regression,
predictor_equal,
)
from xgboost.testing.data import np_dtypes
from xgboost.testing.data import check_inf, np_dtypes
class TestQuantileDMatrix:
@@ -244,6 +244,10 @@ class TestQuantileDMatrix:
from_dm = xgb.QuantileDMatrix(X, weight=w, ref=Xy)
assert predictor_equal(from_qdm, from_dm)
def test_check_inf(self) -> None:
rng = np.random.default_rng(1994)
check_inf(rng)
# we don't test empty Quantile DMatrix in single node construction.
@given(
strategies.integers(1, 1000),