Convert numpy float to Python float in feat score. (#7047)

This commit is contained in:
Jiaming Yuan
2021-06-21 20:58:43 +08:00
committed by GitHub
parent bbfffb444d
commit da1ad798ca
2 changed files with 6 additions and 1 deletions

View File

@@ -171,6 +171,11 @@ class TestBasic:
with pytest.raises(ValueError):
booster.get_fscore()
booster.feature_names = None
# Use JSON to make sure the output has native Python type
scores = json.loads(json.dumps(booster.get_fscore()))
np.testing.assert_allclose(scores["f0"], 6.0)
def test_load_file_invalid(self):
with pytest.raises(xgb.core.XGBoostError):
xgb.Booster(model_file='incorrect_path')