diff --git a/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaInputTest.java b/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaInputTest.java index 3760b58a4a..d4e95a7244 100644 --- a/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaInputTest.java +++ b/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaInputTest.java @@ -44,7 +44,7 @@ public void setup() { TextEmbeddingModelConfig.PoolingMode.MEAN, true, 512); mLRegisterModelMetaInput = new MLRegisterModelMetaInput("Model Name", FunctionName.BATCH_RCF, "model_group_id", "1.0", "Model Description", null, null, MLModelFormat.TORCH_SCRIPT, MLModelState.DEPLOYING, - 200L, "123", config, 2, + 200L, "123", config, null, 2, null, null, false, false, false); } diff --git a/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaRequestTest.java b/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaRequestTest.java index 39243a887a..433ce3acdb 100644 --- a/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaRequestTest.java +++ b/common/src/test/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaRequestTest.java @@ -33,7 +33,7 @@ public void setUp() { config = new TextEmbeddingModelConfig("Model Type", 123, FrameworkType.SENTENCE_TRANSFORMERS, "All Config", TextEmbeddingModelConfig.PoolingMode.MEAN, true, 512); mlRegisterModelMetaInput = new MLRegisterModelMetaInput("Model Name", FunctionName.BATCH_RCF, "Model Group Id", "1.0", - "Model Description", null, null, MLModelFormat.TORCH_SCRIPT, MLModelState.DEPLOYING, 200L, "123", config, 2, null, null, null, null, null); + "Model Description", null, null, MLModelFormat.TORCH_SCRIPT, MLModelState.DEPLOYING, 200L, "123", config, null, 2, null, null, null, null, null); } @Test diff --git a/plugin/src/main/java/org/opensearch/ml/model/MLModelManager.java b/plugin/src/main/java/org/opensearch/ml/model/MLModelManager.java index d4c1429e62..b2c7e94981 100644 --- a/plugin/src/main/java/org/opensearch/ml/model/MLModelManager.java +++ b/plugin/src/main/java/org/opensearch/ml/model/MLModelManager.java @@ -298,7 +298,7 @@ private void uploadMLModelMeta(MLRegisterModelMetaInput mlRegisterModelMetaInput .modelFormat(mlRegisterModelMetaInput.getModelFormat()) .modelState(MLModelState.REGISTERING) .modelConfig(mlRegisterModelMetaInput.getModelConfig()) - .deploySetting(mlRegisterModelMetaInput.getdeploySetting()) + .deploySetting(mlRegisterModelMetaInput.getDeploySetting()) .totalChunks(mlRegisterModelMetaInput.getTotalChunks()) .modelContentHash(mlRegisterModelMetaInput.getModelContentHashValue()) .modelContentSizeInBytes(mlRegisterModelMetaInput.getModelContentSizeInBytes()) @@ -533,7 +533,7 @@ private void indexRemoteModel( .connector(registerModelInput.getConnector()) .connectorId(registerModelInput.getConnectorId()) .modelConfig(registerModelInput.getModelConfig()) - .deploySetting(registerModelInput.getdeploySetting()) + .deploySetting(registerModelInput.getDeploySetting()) .createdTime(now) .lastUpdateTime(now) .isHidden(registerModelInput.getIsHidden()) @@ -600,7 +600,7 @@ void indexRemoteModel(MLRegisterModelInput registerModelInput, MLTask mlTask, St .connector(registerModelInput.getConnector()) .connectorId(registerModelInput.getConnectorId()) .modelConfig(registerModelInput.getModelConfig()) - .deploySetting(registerModelInput.getdeploySetting()) + .deploySetting(registerModelInput.getDeploySetting()) .createdTime(now) .lastUpdateTime(now) .isHidden(registerModelInput.getIsHidden()) @@ -667,7 +667,7 @@ private void registerModelFromUrl(MLRegisterModelInput registerModelInput, MLTas .modelFormat(registerModelInput.getModelFormat()) .modelState(MLModelState.REGISTERING) .modelConfig(registerModelInput.getModelConfig()) - .deploySetting(registerModelInput.getdeploySetting()) + .deploySetting(registerModelInput.getDeploySetting()) .createdTime(now) .lastUpdateTime(now) .isHidden(registerModelInput.getIsHidden())