[CI] Refactor tests to reduce CI time. (#8312)

This commit is contained in:
Rory Mitchell
2022-10-12 11:32:06 +02:00
committed by GitHub
parent 39afdac3be
commit ce0382dcb0
19 changed files with 95 additions and 122 deletions

View File

@@ -6,10 +6,12 @@ import shutil
import urllib.request
import zipfile
import sys
import pytest
sys.path.append("tests/python")
import testing as tm # noqa
pytestmark = pytest.mark.timeout(10)
class TestRanking:
@classmethod
@@ -96,7 +98,7 @@ class TestRanking:
# specify validations set to watch performance
watchlist = [(cls.dtest, 'eval'), (cls.dtrain, 'train')]
num_trees = 2500
num_trees = 100
check_metric_improvement_rounds = 10
evals_result = {}