Retrieve remote model id from registration response in IT to avoid fl…

…aky (#3244)

Signed-off-by: zane-neo <zaniu@amazon.com>
(cherry picked from commit 1d30671)

plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java

@@ -987,15 +987,11 @@ public String registerRemoteModel(String createConnectorInput, String modelName,
         String connectorId = (String) responseMap.get("connector_id");
         response = RestMLRemoteInferenceIT.registerRemoteModel(modelName, modelName, connectorId);
         responseMap = parseResponseToMap(response);
-        String taskId = (String) responseMap.get("task_id");
-        waitForTask(taskId, MLTaskState.COMPLETED);
-        response = RestMLRemoteInferenceIT.getTask(taskId);
-        responseMap = parseResponseToMap(response);
         String modelId = (String) responseMap.get("model_id");
         if (deploy) {
             response = RestMLRemoteInferenceIT.deployRemoteModel(modelId);
             responseMap = parseResponseToMap(response);
-            taskId = (String) responseMap.get("task_id");
+            String taskId = (String) responseMap.get("task_id");
             waitForTask(taskId, MLTaskState.COMPLETED);
         }
         return modelId;