From 5792d3a23890edaa5d32b0f6bfc726c3e9956f3a Mon Sep 17 00:00:00 2001 From: Jon Bratseth Date: Wed, 3 Apr 2019 21:30:28 +0200 Subject: Add tensor value type --- .../src/main/java/com/yahoo/document/update/TensorModifyUpdate.java | 2 +- .../src/main/java/com/yahoo/document/update/TensorRemoveUpdate.java | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'document') diff --git a/document/src/main/java/com/yahoo/document/update/TensorModifyUpdate.java b/document/src/main/java/com/yahoo/document/update/TensorModifyUpdate.java index 2773f9d31da..435c8fcdc65 100644 --- a/document/src/main/java/com/yahoo/document/update/TensorModifyUpdate.java +++ b/document/src/main/java/com/yahoo/document/update/TensorModifyUpdate.java @@ -38,7 +38,7 @@ public class TensorModifyUpdate extends ValueUpdate { * Converts the given tensor type to a type that is compatible for being used in this update (has only mapped dimensions). */ public static TensorType convertDimensionsToMapped(TensorType type) { - TensorType.Builder builder = new TensorType.Builder(); + TensorType.Builder builder = new TensorType.Builder(type.valueType()); type.dimensions().stream().forEach(dim -> builder.mapped(dim.name())); return builder.build(); } diff --git a/document/src/main/java/com/yahoo/document/update/TensorRemoveUpdate.java b/document/src/main/java/com/yahoo/document/update/TensorRemoveUpdate.java index 335cda8e133..981120af145 100644 --- a/document/src/main/java/com/yahoo/document/update/TensorRemoveUpdate.java +++ b/document/src/main/java/com/yahoo/document/update/TensorRemoveUpdate.java @@ -97,7 +97,7 @@ public class TensorRemoveUpdate extends ValueUpdate { } public static TensorType extractSparseDimensions(TensorType type) { - TensorType.Builder builder = new TensorType.Builder(); + TensorType.Builder builder = new TensorType.Builder(type.valueType()); type.dimensions().stream().filter(dim -> ! dim.isIndexed()).forEach(dim -> builder.mapped(dim.name())); return builder.build(); } -- cgit v1.2.3