1 files changed, 7 insertions, 1 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java b/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java
index d548b13601f..036f5e3ee5d 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/TensorType.java
@@ -29,10 +29,16 @@ public class TensorType {
     /** The empty tensor type - which is the same as a double */
     public static final TensorType empty = new TensorType(ValueType.DOUBLE, Collections.emptyList());
 
-    private final ValueType valueType;
+    private ValueType valueType;
 
     public final ValueType valueType() { return valueType; }
 
+    //TODO Remove once value type is wired in were it should.
+    public final TensorType valueType(ValueType valueType) {
+        this.valueType = valueType;
+        return this;
+    }
+
     /** Sorted list of the dimensions of this */
     private final ImmutableList<Dimension> dimensions;