File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plan/logical/inference Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -137,8 +137,8 @@ protected AttributeSet computeReferences() {
137137 }
138138
139139 public static AttributeSet computeReferences (List <Alias > fields ) {
140- AttributeSet generated = new AttributeSet (asAttributes (fields ));
141- return Expressions .references (fields ).subtract (generated );
140+ AttributeSet rerankFields = new AttributeSet (asAttributes (fields ));
141+ return Expressions .references (fields ).subtract (rerankFields );
142142 }
143143
144144 @ Override
@@ -185,6 +185,6 @@ public List<Attribute> output() {
185185 }
186186
187187 public static boolean planHasAttribute (QueryPlan <?> plan , Attribute attribute ) {
188- return plan .output ().stream ().anyMatch (attr -> attr .equals (attribute ));
188+ return plan .outputSet ().stream ().anyMatch (attr -> attr .equals (attribute ));
189189 }
190190}
You can’t perform that action at this time.
0 commit comments