From bcbc8dbe7fc1c470684b23ac11e6409c7c4a356f Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Wed, 2 Oct 2024 12:30:30 -0700 Subject: [PATCH] Fix typo --- qa/L0_e2e/generate_example_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qa/L0_e2e/generate_example_model.py b/qa/L0_e2e/generate_example_model.py index a8287d8..c5f6c12 100644 --- a/qa/L0_e2e/generate_example_model.py +++ b/qa/L0_e2e/generate_example_model.py @@ -311,7 +311,7 @@ def serialize_model(model, directory, output_format="xgboost"): model.save_model(model_path) new_model_path = os.path.join(directory, "xgboost.model") os.rename(model_path, new_model_path) - return model_path + return new_model_path if output_format == "xgboost_json": model_path = os.path.join(directory, "xgboost.json") model.save_model(model_path)