From 148c25f55224d0bcbe5805fc5f360a21d294b676 Mon Sep 17 00:00:00 2001
From: zane-neo <zaniu@amazon.com>
Date: Tue, 11 Jun 2024 08:58:43 +0800
Subject: [PATCH] Update
 plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java

Co-authored-by: Yaliang Wu <ylwu@amazon.com>
Signed-off-by: zane-neo <zaniu@amazon.com>
---
 .../test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java b/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
index 60801cc28a..3460c20031 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
@@ -911,7 +911,7 @@ public Map predictTextEmbedding(String modelId) throws IOException {
         return result;
     }
 
-    public Map predictRemoteModel(String modelId, MLInput input) throws IOException {
+    public Map predictTextEmbeddingModel(String modelId, MLInput input) throws IOException {
         String requestBody = TestHelper.toJsonString(input);
         Response response = TestHelper
             .makeRequest(client(), "POST", "/_plugins/_ml/_predict/TEXT_EMBEDDING/" + modelId, null, requestBody, null);