From 3e4a30511e63ec3b4aa23021d9ba90ac8e000d42 Mon Sep 17 00:00:00 2001 From: Sicheng Song Date: Fri, 28 Feb 2025 21:47:49 +0000 Subject: [PATCH] spotless Signed-off-by: Sicheng Song --- .../org/opensearch/ml/rest/RestMLPredictionActionTests.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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