From 1e907ff8f8757db5db48a8524893bf1300c69388 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Wed, 2 Oct 2024 13:21:56 -0700 Subject: [PATCH] Update train script --- 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 c5f6c12..942cfc7 100644 --- a/qa/L0_e2e/generate_example_model.py +++ b/qa/L0_e2e/generate_example_model.py @@ -551,7 +551,7 @@ def parse_args(): ) parser.add_argument( "--format", - choices=("xgboost", "xgboost_json", "lightgbm", "pickle"), + choices=("xgboost", "xgboost_json", "xgboost_ubj", "lightgbm", "pickle"), default=None, help="serialization format for model", )