aboutsummaryrefslogtreecommitdiffstats
path: root/model-integration
diff options
context:
space:
mode:
authorHenning Baldersheim <balder@yahoo-inc.com>2023-01-17 21:59:41 +0100
committerHenning Baldersheim <balder@yahoo-inc.com>2023-01-17 22:08:42 +0100
commit88495645019196f2d71e38b82f6ef71e6f2a81bb (patch)
tree14b9fd8a0fc01d6b89a1f99091a511e144f5f1d6 /model-integration
parent6c4c57469342d4d28bdebef9e38d45ec8281c99c (diff)
Unify on Streams.toList()
Diffstat (limited to 'model-integration')
-rw-r--r--model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java3
-rw-r--r--model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Squeeze.java4
2 files changed, 3 insertions, 4 deletions
diff --git a/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java b/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java
index 2b9e3a2ab60..81150fe99b0 100644
--- a/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java
+++ b/model-integration/src/main/java/ai/vespa/embedding/huggingface/HuggingFaceEmbedder.java
@@ -14,7 +14,6 @@ import com.yahoo.embedding.huggingface.HuggingFaceEmbedderConfig;
import java.io.*;
import java.nio.file.Paths;
import java.util.*;
-import java.util.stream.Collectors;
import org.slf4j.LoggerFactory;
import org.slf4j.Logger;
@@ -87,7 +86,7 @@ public class HuggingFaceEmbedder implements Embedder {
public List<Integer> longToInteger(long[] values) {
return Arrays.stream(values)
.boxed().map(Long::intValue)
- .collect(Collectors.toList());
+ .toList();
}
@Override
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Squeeze.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Squeeze.java
index 9229d6af254..9c797a990bc 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Squeeze.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Squeeze.java
@@ -39,14 +39,14 @@ public class Squeeze extends IntermediateOperation {
squeezeDimensions = inputType.type().dimensions().stream().
filter(dim -> OrderedTensorType.dimensionSize(dim) == 1).
map(TensorType.Dimension::name).
- collect(Collectors.toList());
+ toList();
} else {
squeezeDimensions = squeezeDimsAttr.get().stream().map(Value::asDouble).map(Double::intValue).
map(i -> i < 0 ? inputType.type().dimensions().size() - i : i).
map(i -> inputType.type().dimensions().get(i)).
filter(dim -> OrderedTensorType.dimensionSize(dim) == 1).
map(TensorType.Dimension::name).
- collect(Collectors.toList());
+ toList();
}
return squeezeDimensions.isEmpty() ? inputType : reducedType(inputType);