format code

Signed-off-by: zane-neo <zaniu@amazon.com>
zane-neo · Mar 3, 2025 · ef3e22c · ef3e22c
1 parent 8ec6f7e
commit ef3e22c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java
@@ -302,7 +302,9 @@ private void validateOutput(String errorMsg, Map<String, Object> output, String
         List outputList = (List) output.get("output");
         assertEquals(errorMsg, 1, outputList.size());
         assertTrue(errorMsg, outputList.get(0) instanceof Map);
-        String typeErrorMsg = errorMsg + " first element in the output list is type of: " + ((Map<?, ?>)outputList.get(0)).get("data").getClass().getName();
+        String typeErrorMsg = errorMsg
+            + " first element in the output list is type of: "
+            + ((Map<?, ?>) outputList.get(0)).get("data").getClass().getName();
         assertTrue(typeErrorMsg, ((Map<?, ?>) outputList.get(0)).get("data") instanceof List);
         assertEquals(errorMsg, ((Map<?, ?>) outputList.get(0)).get("data_type"), dataType);
     }

diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java
@@ -88,7 +88,9 @@ private void validateOutput(String errorMsg, Map<String, Object> output, String
         List outputList = (List) output.get("output");
         assertEquals(errorMsg, 2, outputList.size());
         assertTrue(errorMsg, outputList.get(0) instanceof Map);
-        String typeErrorMsg = errorMsg + " first element in the output list is type of: " + ((Map<?, ?>)outputList.get(0)).get("data").getClass().getName();
+        String typeErrorMsg = errorMsg
+            + " first element in the output list is type of: "
+            + ((Map<?, ?>) outputList.get(0)).get("data").getClass().getName();
         assertTrue(typeErrorMsg, ((Map<?, ?>) outputList.get(0)).get("data") instanceof List);
         assertTrue(errorMsg, ((Map<?, ?>) outputList.get(0)).get("data_type").equals(dataType));
     }