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 107c132fe1..d57f214609 100644 --- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java +++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLRemoteInferenceIT.java @@ -131,7 +131,6 @@ public void testDeployRemoteModel() throws IOException, InterruptedException { waitForTask(taskId, MLTaskState.COMPLETED); } - @Ignore public void testPredictRemoteModel() throws IOException, InterruptedException { Response response = createConnector(completionModelConnectorEntity); Map responseMap = parseResponseToMap(response); @@ -184,7 +183,6 @@ public void testUndeployRemoteModel() throws IOException, InterruptedException { assertTrue(responseMap.toString().contains("undeployed")); } - @Ignore public void testOpenAIChatCompletionModel() throws IOException, InterruptedException { String entity = "{\n" + " \"name\": \"OpenAI chat model Connector\",\n" @@ -241,7 +239,6 @@ public void testOpenAIChatCompletionModel() throws IOException, InterruptedExcep assertNotNull(responseMap); } - @Ignore public void testOpenAIEditsModel() throws IOException, InterruptedException { String entity = "{\n" + " \"name\": \"OpenAI Edit model Connector\",\n" @@ -307,7 +304,6 @@ public void testOpenAIEditsModel() throws IOException, InterruptedException { assertFalse(((String) responseMap.get("text")).isEmpty()); } - @Ignore public void testOpenAIModerationsModel() throws IOException, InterruptedException { String entity = "{\n" + " \"name\": \"OpenAI moderations model Connector\",\n"