fix typo error

Signed-off-by: xinyual <[email protected]>
opensearch-project · Sep 26, 2023 · dca433a · dca433a
1 parent a837080
commit dca433a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/...thms/src/main/java/org/opensearch/ml/engine/algorithms/SentenceTransformerTranslator.java b/...thms/src/main/java/org/opensearch/ml/engine/algorithms/SentenceTransformerTranslator.java
@@ -23,7 +23,7 @@
 import java.util.List;
 import java.util.Map;
 
-abstract public class SentenceTransformerTranslator implements ServingTranslator {
+public abstract class SentenceTransformerTranslator implements ServingTranslator {
     protected HuggingFaceTokenizer tokenizer;
 
     @Override

diff --git a/...in/java/org/opensearch/ml/engine/algorithms/sparse_encoding/SparseEncodingTranslator.java b/...in/java/org/opensearch/ml/engine/algorithms/sparse_encoding/SparseEncodingTranslator.java
@@ -29,19 +29,6 @@
 import static org.opensearch.ml.common.CommonValue.ML_MAP_RESPONSE_KEY;
 
 public class SparseEncodingTranslator extends SentenceTransformerTranslator {
-    private Map<String, Float>  convertOutput(NDArray array)
-    {
-        Map<String, Float> map = new HashMap<>();
-        NDArray nonZeroIndices = array.nonzero().squeeze();
-
-        for (long index : nonZeroIndices.toLongArray()) {
-            String s = this.tokenizer.decode(new long[]{index}, true);
-            if (!s.isEmpty()){
-                map.put(s, array.getFloat(index));
-            }
-        }
-        return map;
-    }
     @Override
     public Output processOutput(TranslatorContext ctx, NDList list) {
         Output output = new Output(200, "OK");
@@ -64,4 +51,17 @@ public Output processOutput(TranslatorContext ctx, NDList list) {
         output.add(modelTensorOutput.toBytes());
         return output;
     }
+    private Map<String, Float>  convertOutput(NDArray array)
+    {
+        Map<String, Float> map = new HashMap<>();
+        NDArray nonZeroIndices = array.nonzero().squeeze();
+
+        for (long index : nonZeroIndices.toLongArray()) {
+            String s = this.tokenizer.decode(new long[]{index}, true);
+            if (!s.isEmpty()){
+                map.put(s, array.getFloat(index));
+            }
+        }
+        return map;
+    }
 }