diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java index 271a737f8a..309467f5c3 100644 --- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java +++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java @@ -739,8 +739,7 @@ public void testCohereGenerateTextModel() throws IOException, InterruptedExcepti responseMap = (Map) responseList.get(0); assertFalse(((String) responseMap.get("text")).isEmpty()); } - - @Ignore + public void testCohereClassifyModel() throws IOException, InterruptedException { // Skip test if key is null if (COHERE_KEY == null) {