Skip to content

Commit b7e0aa4

Browse files
committed
Fix constants bug
1 parent 50a2528 commit b7e0aa4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pkg/operator/api/context/dependencies.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ func (ctx *Context) aggregatesDependencies(aggregate *Aggregate) strset.Set {
101101
dependencies.Add(pythonPackage.GetID())
102102
}
103103

104-
for _, res := range ctx.ExtractCortexResources(aggregate.Input, resource.ConstantType, resource.RawColumnType) {
104+
for _, res := range ctx.ExtractCortexResources(aggregate.Input, resource.RawColumnType) {
105105
dependencies.Add(res.GetID())
106106
}
107107

@@ -115,7 +115,7 @@ func (ctx *Context) transformedColumnDependencies(transformedColumn *Transformed
115115
dependencies.Add(pythonPackage.GetID())
116116
}
117117

118-
for _, res := range ctx.ExtractCortexResources(transformedColumn.Input, resource.ConstantType, resource.RawColumnType, resource.AggregateType) {
118+
for _, res := range ctx.ExtractCortexResources(transformedColumn.Input, resource.RawColumnType, resource.AggregateType) {
119119
dependencies.Add(res.GetID())
120120
}
121121

@@ -143,7 +143,7 @@ func (ctx *Context) modelDependencies(model *Model) strset.Set {
143143
dependencies.Add(model.Dataset.ID)
144144

145145
combinedInput := []interface{}{model.Input, model.TrainingInput, model.TargetColumn}
146-
for _, res := range ctx.ExtractCortexResources(combinedInput, resource.ConstantType, resource.RawColumnType, resource.AggregateType, resource.TransformedColumnType) {
146+
for _, res := range ctx.ExtractCortexResources(combinedInput, resource.RawColumnType, resource.AggregateType, resource.TransformedColumnType) {
147147
dependencies.Add(res.GetID())
148148
}
149149

0 commit comments

Comments
 (0)