[AINode] Fix call inference error (#17308)

Author: CRZbulabula

iotdb-core/node-commons/src/main/java/org/apache/iotdb/commons/model/ModelInformation.java

@@ -35,8 +35,6 @@ public class ModelInformation {
   private static final int[] DEFAULT_MODEL_INPUT_SHAPE = new int[] {2880, 1};
   private static final int[] DEFAULT_MODEL_OUTPUT_SHAPE = new int[] {720, 1};
 
-  ModelType modelType;
-
   private final String modelId;
 
   private final int[] inputShape;
@@ -60,15 +58,13 @@ public ModelInformation(String modelId) {
   }
 
   public ModelInformation(
-      ModelType modelType,
       String modelId,
       int[] inputShape,
       int[] outputShape,
       TSDataType[] inputDataType,
       TSDataType[] outputDataType,
       String attribute,
       ModelStatus status) {
-    this.modelType = modelType;
     this.modelId = modelId;
     this.inputShape = inputShape;
     this.outputShape = outputShape;
@@ -85,7 +81,6 @@ public ModelInformation(
       TSDataType[] inputDataType,
       TSDataType[] outputDataType,
       String attribute) {
-    this.modelType = ModelType.USER_DEFINED;
     this.modelId = modelId;
     this.inputShape = inputShape;
     this.outputShape = outputShape;
@@ -95,7 +90,6 @@ public ModelInformation(
   }
 
   public ModelInformation(String modelId, ModelStatus status) {
-    this.modelType = ModelType.BUILT_IN_FORECAST;
     this.modelId = modelId;
     this.inputShape = new int[0];
     this.outputShape = new int[0];
@@ -104,21 +98,6 @@ public ModelInformation(String modelId, ModelStatus status) {
     this.status = status;
   }
 
-  // init built-in modelInformation
-  public ModelInformation(ModelType modelType, String modelId) {
-    this.modelType = modelType;
-    this.modelId = modelId;
-    this.inputShape = new int[2];
-    this.outputShape = new int[2];
-    this.inputDataType = new TSDataType[0];
-    this.outputDataType = new TSDataType[0];
-    this.status = ModelStatus.ACTIVE;
-  }
-
-  public boolean isBuiltIn() {
-    return modelType != ModelType.USER_DEFINED;
-  }
-
   public boolean available() {
     return status == ModelStatus.ACTIVE;
   }
@@ -139,26 +118,6 @@ public void setOutputLength(int length) {
     outputShape[0] = length;
   }
 
-  // calculation modelType and outputColumn metadata for different built-in models
-  public void setInputColumnSize(int size) {
-    inputShape[1] = size;
-    if (modelType == ModelType.BUILT_IN_FORECAST) {
-      outputShape[1] = size;
-    } else if (modelType == ModelType.BUILT_IN_ANOMALY_DETECTION) {
-      outputShape[1] = 1;
-    } else {
-      outputShape[1] = size;
-    }
-    if (modelType == ModelType.BUILT_IN_FORECAST) {
-      buildOutputDataTypeForBuiltInModel(TSDataType.DOUBLE, outputShape[1]);
-    } else if (modelType == ModelType.BUILT_IN_ANOMALY_DETECTION) {
-      buildOutputDataTypeForBuiltInModel(TSDataType.INT32, outputShape[1]);
-    } else {
-      buildOutputDataTypeForBuiltInModel(TSDataType.FLOAT, outputShape[1]);
-      buildInputDataTypeForBuiltInModel(TSDataType.FLOAT, inputShape[1]);
-    }
-  }
-
   public void setInputDataType(TSDataType[] inputDataType) {
     this.inputDataType = inputDataType;
   }
@@ -206,7 +165,6 @@ public void setAttribute(String attribute) {
   }
 
   public void serialize(DataOutputStream stream) throws IOException {
-    ReadWriteIOUtils.write(modelType.ordinal(), stream);
     ReadWriteIOUtils.write(status.ordinal(), stream);
     ReadWriteIOUtils.write(modelId, stream);
     if (status == ModelStatus.UNAVAILABLE) {
@@ -231,7 +189,6 @@ public void serialize(DataOutputStream stream) throws IOException {
   }
 
   public void serialize(FileOutputStream stream) throws IOException {
-    ReadWriteIOUtils.write(modelType.ordinal(), stream);
     ReadWriteIOUtils.write(status.ordinal(), stream);
     ReadWriteIOUtils.write(modelId, stream);
     if (status == ModelStatus.UNAVAILABLE) {
@@ -256,7 +213,6 @@ public void serialize(FileOutputStream stream) throws IOException {
   }
 
   public void serialize(ByteBuffer byteBuffer) {
-    ReadWriteIOUtils.write(modelType.ordinal(), byteBuffer);
     ReadWriteIOUtils.write(status.ordinal(), byteBuffer);
     ReadWriteIOUtils.write(modelId, byteBuffer);
     if (status == ModelStatus.UNAVAILABLE) {
@@ -281,7 +237,6 @@ public void serialize(ByteBuffer byteBuffer) {
   }
 
   public static ModelInformation deserialize(ByteBuffer buffer) {
-    ModelType modelType = ModelType.values()[ReadWriteIOUtils.readInt(buffer)];
     ModelStatus status = ModelStatus.values()[ReadWriteIOUtils.readInt(buffer)];
     String modelName = ReadWriteIOUtils.readString(buffer);
     if (status == ModelStatus.UNAVAILABLE) {
@@ -311,18 +266,10 @@ public static ModelInformation deserialize(ByteBuffer buffer) {
     String attribute = ReadWriteIOUtils.readString(buffer);
 
     return new ModelInformation(
-        modelType,
-        modelName,
-        inputShape,
-        outputShape,
-        inputDataType,
-        outputDataType,
-        attribute,
-        status);
+        modelName, inputShape, outputShape, inputDataType, outputDataType, attribute, status);
   }
 
   public static ModelInformation deserialize(InputStream stream) throws IOException {
-    ModelType modelType = ModelType.values()[ReadWriteIOUtils.readInt(stream)];
     ModelStatus status = ModelStatus.values()[ReadWriteIOUtils.readInt(stream)];
     String modelName = ReadWriteIOUtils.readString(stream);
     if (status == ModelStatus.UNAVAILABLE) {
@@ -351,21 +298,13 @@ public static ModelInformation deserialize(InputStream stream) throws IOExceptio
 
     String attribute = ReadWriteIOUtils.readString(stream);
     return new ModelInformation(
-        modelType,
-        modelName,
-        inputShape,
-        outputShape,
-        inputDataType,
-        outputDataType,
-        attribute,
-        status);
+        modelName, inputShape, outputShape, inputDataType, outputDataType, attribute, status);
   }
 
   public ByteBuffer serializeShowModelResult() throws IOException {
     PublicBAOS buffer = new PublicBAOS();
     DataOutputStream stream = new DataOutputStream(buffer);
     ReadWriteIOUtils.write(modelId, stream);
-    ReadWriteIOUtils.write(modelType.toString(), stream);
     ReadWriteIOUtils.write(status.toString(), stream);
     ReadWriteIOUtils.write(Arrays.toString(inputShape), stream);
     ReadWriteIOUtils.write(Arrays.toString(outputShape), stream);
@@ -382,7 +321,6 @@ public boolean equals(Object obj) {
     if (obj instanceof ModelInformation) {
       ModelInformation other = (ModelInformation) obj;
       return modelId.equals(other.modelId)
-          && modelType.equals(other.modelType)
           && Arrays.equals(inputShape, other.inputShape)
           && Arrays.equals(outputShape, other.outputShape)
           && Arrays.equals(inputDataType, other.inputDataType)