Remove experimental_json_serialization from tests. (#6640)

2021-01-27 17:44:49 +08:00
parent 8968ca7c0a
commit bc08e0c9d1
5 changed files with 9 additions and 22 deletions
--- a/tests/python-gpu/test_gpu_prediction.py
+++ b/tests/python-gpu/test_gpu_prediction.py
@@ -283,9 +283,10 @@ class TestGPUPredict:
        y = (x0 * 10 - 20) + (x1 - 2)
        dtrain = xgb.DMatrix(df, label=y, enable_categorical=True)

-        params = {'tree_method': 'gpu_hist', 'predictor': 'gpu_predictor',
-                  'enable_experimental_json_serialization': True,
-                  'max_depth': 3, 'learning_rate': 1.0, 'base_score': 0.0, 'eval_metric': 'rmse'}
+        params = {
+            'tree_method': 'gpu_hist', 'predictor': 'gpu_predictor',
+            'max_depth': 3, 'learning_rate': 1.0, 'base_score': 0.0, 'eval_metric': 'rmse'
+        }

        eval_history = {}
        bst = xgb.train(params, dtrain, num_boost_round=5, evals=[(dtrain, 'train')],