Fix compilation

Signed-off-by: Sicheng Song <sicheng.song@outlook.com>
opensearch-project · Mar 26, 2024 · abb2173 · abb2173
1 parent 71af4ea
commit abb2173
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/...st/java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaInputTest.java b/...st/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/.../java/org/opensearch/ml/common/transport/upload_chunk/MLRegisterModelMetaRequestTest.java b/.../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
@@ -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())