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:
Jiaming Yuan
2022-10-26 16:56:11 +08:00
committed by GitHub
parent 7e53189e7c
commit cf70864fa3
66 changed files with 652 additions and 595 deletions

View File

@@ -3,20 +3,17 @@ import json
import os
import pickle
import subprocess
import sys
import numpy as np
import pytest
import xgboost as xgb
from xgboost import XGBClassifier, testing
sys.path.append("tests/python")
import testing as tm
from xgboost import XGBClassifier
from xgboost import testing as tm
model_path = './model.pkl'
pytestmark = testing.timeout(30)
pytestmark = tm.timeout(30)
def build_dataset():