diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLPredictionActionTests.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLPredictionActionTests.java index 5fb0239ac6..9028ce174e 100644 --- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLPredictionActionTests.java +++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLPredictionActionTests.java @@ -127,7 +127,8 @@ public void testRoutes_Batch() { @Test public void testGetRequest() throws IOException { RestRequest request = getRestRequest_PredictModel(); - MLPredictionTaskRequest mlPredictionTaskRequest = restMLPredictionAction.getRequest("modelId", FunctionName.KMEANS.name(), FunctionName.KMEANS.name(), request); + MLPredictionTaskRequest mlPredictionTaskRequest = restMLPredictionAction + .getRequest("modelId", FunctionName.KMEANS.name(), FunctionName.KMEANS.name(), request); MLInput mlInput = mlPredictionTaskRequest.getMlInput(); verifyParsedKMeansMLInput(mlInput); @@ -140,7 +141,8 @@ public void testGetRequest_RemoteInferenceDisabled() throws IOException { when(mlFeatureEnabledSetting.isRemoteInferenceEnabled()).thenReturn(false); RestRequest request = getRestRequest_PredictModel(); - MLPredictionTaskRequest mlPredictionTaskRequest = restMLPredictionAction.getRequest("modelId", FunctionName.REMOTE.name(), "text_embedding", request); + MLPredictionTaskRequest mlPredictionTaskRequest = restMLPredictionAction + .getRequest("modelId", FunctionName.REMOTE.name(), "text_embedding", request); } @Test