diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2023-01-17 21:59:41 +0100 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2023-01-17 22:08:42 +0100 |
commit | 88495645019196f2d71e38b82f6ef71e6f2a81bb (patch) | |
tree | 14b9fd8a0fc01d6b89a1f99091a511e144f5f1d6 /model-integration/src/main/java | |
parent | 6c4c57469342d4d28bdebef9e38d45ec8281c99c (diff) |
Unify on Streams.toList()
Diffstat (limited to 'model-integration/src/main/java')
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); |