java-topology/defects/flink/patch/flink-0007-ml-predict-tablefieldnames-hashset.patch

12 lines
860 B
Diff

# UNDF: UNDF-2026-000000762
# UNDF: (leave blank)
--- a/flink-table/flink-table-common/src/main/java/org/apache/flink/table/types/inference/strategies/MLPredictTypeStrategy.java
+++ b/flink-table/flink-table-common/src/main/java/org/apache/flink/table/types/inference/strategies/MLPredictTypeStrategy.java
@@ -165,7 +165,7 @@ class MLPredictTypeStrategy implements TypeStrategy {
// Check that descriptor column names exist in table columns
- List<String> tableFieldNames = DataType.getFieldNames(tableSemantics.dataType());
+ Set<String> tableFieldNames = new HashSet<>(DataType.getFieldNames(tableSemantics.dataType()));
List<String> descriptorColumnNames = descriptorColumns.getNames();
for (String descriptorColumnName : descriptorColumnNames) {
if (!tableFieldNames.contains(descriptorColumnName)) {