Apply feedback

nathaliellenaa · nathaliellenaa · commit fb59fb3ace73 · 2025-03-17T15:21:50.000-07:00
Signed-off-by: Nathalie Jonathan &lt;nathhjo@amazon.com&gt;
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java
@@ -78,7 +78,7 @@ public void test_cohereInference_withDifferent_postProcessFunction() {
             List output = (List) inferenceResult.get("inference_results");
             assertEquals(errorMsg, 1, output.size());
             assertTrue(errorMsg, output.get(0) instanceof Map);
-            if (output != null && DATA_TYPE != null) {
+            if (output != null) {
                 validateOutput(errorMsg, (Map) output.get(0), DATA_TYPE.get(postProcessFunction));
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ public void test_cohereInference_withDifferent_postProcessFunction() {`
`78`	`78`	`List output = (List) inferenceResult.get("inference_results");`
`79`	`79`	`assertEquals(errorMsg, 1, output.size());`
`80`	`80`	`assertTrue(errorMsg, output.get(0) instanceof Map);`
`81`		`- if (output != null && DATA_TYPE != null) {`
	`81`	`+ if (output != null) {`
`82`	`82`	`validateOutput(errorMsg, (Map) output.get(0), DATA_TYPE.get(postProcessFunction));`
`83`	`83`	`}`
`84`	`84`	`}`