Fix typo

triton-inference-server · Oct 2, 2024 · bcbc8db · bcbc8db
1 parent 49a5f45
commit bcbc8db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)