Lint.

dmlc · Jan 31, 2021 · df23e8e · df23e8e
1 parent d75cd6d
commit df23e8e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py
@@ -1009,10 +1009,10 @@ def _infer_predict_output(
     test_sample = rng.randn(1, features)
     if inplace:
         # clear the state to avoid gpu_id, gpu_predictor
-        config = booster.save_config()
+        booster_config = booster.save_config()
         booster.set_param({"predictor": "cpu_predictor", "gpu_id": -1})
         test_predt = booster.inplace_predict(test_sample, **kwargs)
-        booster.load_config(config)
+        booster.load_config(booster_config)
     else:
         m = DMatrix(test_sample)
         test_predt = booster.predict(m, **kwargs)