Move Python testing utilities into xgboost module. (#8379)
- Add typehints. - Fixes for pylint. Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
This commit is contained in:
@@ -1,14 +1,16 @@
|
||||
from typing import Dict, List
|
||||
|
||||
import numpy as np
|
||||
import pytest
|
||||
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
|
||||
from xgboost.testing import IteratorForTest, make_batches, non_increasing
|
||||
|
||||
import xgboost as xgb
|
||||
from xgboost import testing
|
||||
from xgboost import testing as tm
|
||||
|
||||
pytestmark = testing.timeout(30)
|
||||
pytestmark = tm.timeout(30)
|
||||
|
||||
|
||||
def test_single_batch(tree_method: str = "approx") -> None:
|
||||
@@ -83,7 +85,7 @@ def run_data_iterator(
|
||||
if tree_method == "gpu_hist":
|
||||
parameters["sampling_method"] = "gradient_based"
|
||||
|
||||
results_from_it: xgb.callback.EvaluationMonitor.EvalsLog = {}
|
||||
results_from_it: Dict[str, Dict[str, List[float]]] = {}
|
||||
from_it = xgb.train(
|
||||
parameters,
|
||||
Xy,
|
||||
@@ -106,7 +108,7 @@ def run_data_iterator(
|
||||
assert Xy.num_row() == n_samples_per_batch * n_batches
|
||||
assert Xy.num_col() == n_features
|
||||
|
||||
results_from_arrays: xgb.callback.EvaluationMonitor.EvalsLog = {}
|
||||
results_from_arrays: Dict[str, Dict[str, List[float]]] = {}
|
||||
from_arrays = xgb.train(
|
||||
parameters,
|
||||
Xy,
|
||||
|
||||
Reference in New Issue
Block a user