Skip to content

Commit b7f839d

Browse files
committed
fix
1 parent 082b4c8 commit b7f839d

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

flink-table/flink-table-common/src/main/java/org/apache/flink/table/functions/ModelTypeUtils.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
package org.apache.flink.table.functions;
2020

21+
import org.apache.flink.annotation.Internal;
2122
import org.apache.flink.table.api.DataTypes;
2223
import org.apache.flink.table.api.DataTypes.Field;
2324
import org.apache.flink.table.api.ValidationException;
@@ -40,6 +41,7 @@
4041
import java.util.Set;
4142
import java.util.stream.Collectors;
4243

44+
@Internal
4345
public class ModelTypeUtils {
4446

4547
public static final InputTypeStrategy ML_PREDICT_INPUT_TYPE_STRATEGY =

flink-table/flink-table-common/src/test/java/org/apache/flink/table/types/inference/strategies/MLPredictInputTypeStrategyTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,6 @@ protected Stream<TestSpec> testData() {
160160
.calledWithModelSemanticsAt(
161161
1, new ModelSemanticsMock(MODEL_INPUT_TYPE, MODEL_OUTPUT_TYPE))
162162
.expectErrorMessage(
163-
"Third argument must be a descriptor with column names for ML_PREDICT function."));
163+
"Third argument must be a descriptor with simple column names for ML_PREDICT function."));
164164
}
165165
}

0 commit comments

Comments
 (0)