Skip to content

Commit 73199b3

Browse files
authored
fix(backend): use json format for xgboost model marshalling (kubeflow#678)
Fixes kubeflow#675 Signed-off-by: Naymul Islam <naymul504@gmail.com>
1 parent d9a4ac7 commit 73199b3

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

examples/serving/xgboost/iris.ipynb

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,9 @@
2121
"cell_type": "code",
2222
"execution_count": null,
2323
"metadata": {
24-
"tags": []
24+
"tags": [
25+
"skip"
26+
]
2527
},
2628
"outputs": [],
2729
"source": [

kale/marshal/backends.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ class XGBoostModelBackend(MarshalBackend):
110110

111111
name = "XGBoost Model backend"
112112
display_name = "xgboost"
113-
file_type = "bst"
113+
file_type = "json"
114114
obj_type_regex = r"xgboost\.core\.Booster"
115115
predictor_type = "xgboost"
116116

0 commit comments

Comments
 (0)