@@ -1,3 +1,4 @@
|
||||
[pytest]
|
||||
markers =
|
||||
mgpu: Mark a test that requires multiple GPUs to run.
|
||||
mgpu: Mark a test that requires multiple GPUs to run.
|
||||
ci: Mark a test that runs only on CI.
|
||||
@@ -4,6 +4,7 @@ import generate_models as gm
|
||||
import json
|
||||
import zipfile
|
||||
import pytest
|
||||
import copy
|
||||
|
||||
|
||||
def run_model_param_check(config):
|
||||
@@ -124,6 +125,9 @@ def test_model_compatibility():
|
||||
if name.startswith('xgboost-'):
|
||||
booster = xgboost.Booster(model_file=path)
|
||||
run_booster_check(booster, name)
|
||||
# Do full serialization.
|
||||
booster = copy.copy(booster)
|
||||
run_booster_check(booster, name)
|
||||
elif name.startswith('xgboost_scikit'):
|
||||
run_scikit_model_check(name, path)
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user