Update plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestC…

…ase.java Co-authored-by: Yaliang Wu <[email protected]> Signed-off-by: zane-neo <[email protected]>
zane-neo · Jun 11, 2024 · 148c25f · 148c25f
1 parent 59fcffb
commit 148c25f
Showing 1 changed file with 1 addition and 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
@@ -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);