remove buffer for sparse encoding output

Signed-off-by: xinyual <[email protected]>
xinyual · Sep 12, 2023 · f98f52a · f98f52a
1 parent 7f0d576
commit f98f52a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
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
@@ -84,11 +84,9 @@ public Output processOutput(TranslatorContext ctx, NDList list) {
             List<Map<String, Float> > listOftokenWeights = new ArrayList<>();
             listOftokenWeights.add(tokenWeightsMap);
             resultMap.put("response", listOftokenWeights);
-            ByteBuffer buffer = ndArray.toByteBuffer();
             ModelTensor tensor = ModelTensor.builder()
                     .name(name)
                     .dataAsMap(resultMap)
-                    .byteBuffer(buffer)
                     .build();
             outputs.add(tensor);
         }

diff --git a/...est/java/org/opensearch/ml/engine/algorithms/sparse_encoding/SparseEncodingModelTest.java b/...est/java/org/opensearch/ml/engine/algorithms/sparse_encoding/SparseEncodingModelTest.java
@@ -140,8 +140,6 @@ public void test_SparseEncoding_Translator_ProcessOutput() throws URISyntaxExcep
         when(ndArray.getFloat(any())).thenReturn(1.0f);
         when(ndArray.toLongArray()).thenReturn(new long[]{10000, 10001});
         when(ndArray.getName()).thenReturn("output");
-        ByteBuffer buffer = mock(MappedByteBuffer.class);
-        when(ndArray.toByteBuffer()).thenReturn(buffer);
         List<NDArray> ndArrayList = Collections.singletonList(ndArray);
         NDList ndList = new NDList(ndArrayList);
         Output output = sparseEncodingTranslator.processOutput(translatorContext, ndList);