Fix compile issue after rebase

Signed-off-by: zane-neo <[email protected]>
zane-neo · Jan 26, 2024 · 8315be5 · 8315be5
1 parent 38cb4ac
commit 8315be5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java
@@ -175,7 +175,7 @@ public static ModelTensors processOutput(
             // in this case, we can use jsonpath to build a List<List<Float>> result from model response.
             if (StringUtils.isBlank(responseFilter))
                 responseFilter = MLPostProcessFunction.getResponseFilter(postProcessFunction);
-            List<List<Float>> vectors = JsonPath.read(modelResponse, responseFilter);
+            List<?> vectors = JsonPath.read(modelResponse, responseFilter);
             List<ModelTensor> processedResponse = executeBuildInPostProcessFunction(
                 vectors,
                 MLPostProcessFunction.get(postProcessFunction)

diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/utils/ScriptUtils.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/utils/ScriptUtils.java
@@ -31,8 +31,8 @@ public static Optional<String> executePreprocessFunction(
     }
 
     public static List<ModelTensor> executeBuildInPostProcessFunction(
-        List<List<Float>> vectors,
-        Function<List<List<Float>>, List<ModelTensor>> function
+        List<?> vectors,
+        Function<List<?>, List<ModelTensor>> function
     ) {
         return function.apply(vectors);
     }