Skip to content

Commit 5fda753

Browse files
author
Yadan Wei
committed
change model load version
1 parent 50b9b96 commit 5fda753

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/integ/test_xgboost.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,10 @@ def test_xgboost_serverless_inference(
137137
model_data=model_data,
138138
role=ROLE,
139139
entry_point=os.path.join(DATA_DIR, "xgboost_abalone", "abalone.py"),
140-
framework_version="1.5-1",
140+
# XGBoost changed its default model format:
141+
# - Old XGBoost (< 2.0): Default save format was binary
142+
# - New XGBoost (>= 2.0): Default save format is JSON/UBJSON
143+
framework_version="3.0-5",
141144
)
142145

143146
xgboost.deploy(

0 commit comments

Comments
 (0)