Update train script

triton-inference-server · Oct 2, 2024 · 1e907ff · 1e907ff
1 parent 845bd0e
commit 1e907ff
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
@@ -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",
     )