Skip to content

Commit 13d9821

Browse files
authored
fix pai xgboost package deps (#2803)
1 parent 29d8df5 commit 13d9821

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

go/codegen/pai/template_tf.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@ oss.save_metas("{{.OSSModelDir}}",
126126
feature_columns_code)
127127
`
128128

129+
// install sklearn-pandas==1.8.0 to fix deps for sklearn2pmml with Python2 on PAI.
129130
const paiRequirementsTmplText = `
130131
adanet==0.8.0
131132
numpy==1.16.2
@@ -136,6 +137,7 @@ shap==0.28.5
136137
scikit-learn==0.20.4
137138
tensorflow-datasets==3.0.0
138139
{{if .IsXGBoost }}
140+
sklearn-pandas==1.8.0
139141
xgboost==0.82
140142
sklearn2pmml==0.56.0
141143
{{end}}

0 commit comments

Comments
 (0)