Skip to content

Commit 564de6a

Browse files
authored
fix typo (#2820)
1 parent 2099ba2 commit 564de6a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

go/codegen/experimental/codegen_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,6 @@ func TestExperimentalXGBCodegen(t *testing.T) {
4343
if err != nil {
4444
t.Errorf("error %s", err)
4545
}
46-
expected := `feature_column_map = {"featuren_columns": [fc.NumericColumn(fd.FieldDesc(name="petal_length", dtype=fd.DataType.FLOAT32, delimiter="", format="", shape=[1], is_sparse=False, vocabulary=[]))]}`
46+
expected := `feature_column_map = {"feature_columns": [fc.NumericColumn(fd.FieldDesc(name="petal_length", dtype=fd.DataType.FLOAT32, delimiter="", format="", shape=[1], is_sparse=False, vocabulary=[]))]}`
4747
a.True(strings.Contains(coulerCode, expected))
4848
}

go/codegen/experimental/xgboost.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ func XGBoostGenerateTrain(trainStmt *ir.TrainStmt, stepIndex int, session *pb.Se
116116
return program.String(), nil
117117
}
118118

119-
var xgbTrainTemplate = `
119+
const xgbTrainTemplate = `
120120
def step_entry_{{.StepIndex}}():
121121
import json
122122
import tempfile
@@ -141,7 +141,7 @@ def step_entry_{{.StepIndex}}():
141141
conn = runtime.db.connect_with_data_source(ds)
142142
143143
{{ if .FeatureColumnCode }}
144-
feature_column_map = {"featuren_columns": [{{.FeatureColumnCode}}]}
144+
feature_column_map = {"feature_columns": [{{.FeatureColumnCode}}]}
145145
{{ else }}
146146
feature_column_map = None
147147
{{ end }}

0 commit comments

Comments
 (0)