fix failure IT

Signed-off-by: zane-neo <[email protected]>
zane-neo · Jun 12, 2024 · 6b0abf7 · 6b0abf7
1 parent bf71571
commit 6b0abf7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java
@@ -234,7 +234,7 @@ public void test_bedrock_multimodal_model_empty_imageInput_null_textInput() thro
             MLInput mlInput = MLInput.builder().inputDataset(inputDataSet).algorithm(FunctionName.TEXT_EMBEDDING).build();
             Map inferenceResult = predictTextEmbeddingModel(modelId, mlInput);
             assertTrue(errorMsg, inferenceResult.containsKey("status"));
-            assertEquals(errorMsg, 400, Integer.parseInt(String.valueOf(inferenceResult.get("status"))));
+            assertTrue(errorMsg, String.valueOf(inferenceResult.get("status")).contains("400"));
             assertTrue(errorMsg, inferenceResult.containsKey("error"));
             assertTrue(errorMsg, inferenceResult.get("error") instanceof Map);
             assertEquals(errorMsg, "illegal_argument_exception", ((Map<?, ?>) inferenceResult.get("error")).get("type"));