Make float builder when appropriate

1 files changed, 2 insertions, 3 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
index 6e587b05460..8e2223def83 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/IndexedTensor.java
@@ -235,8 +235,7 @@ public abstract class IndexedTensor implements Tensor {
             }
 
             if (type.valueType() == TensorType.Value.FLOAT)
-                return new IndexedDoubleTensor.BoundDoubleBuilder(type, sizes);
-                // return new IndexedFloatTensor.BoundFloatBuilder(type, sizes); TODO
+                return new IndexedFloatTensor.BoundFloatBuilder(type, sizes);
             else if (type.valueType() == TensorType.Value.FLOAT)
                 return new IndexedDoubleTensor.BoundDoubleBuilder(type, sizes);
             else
@@ -258,7 +257,7 @@ public abstract class IndexedTensor implements Tensor {
 
         private DimensionSizes sizes;
 
-        static DimensionSizes dimensionSizesOf(TensorType type) {
+        private static DimensionSizes dimensionSizesOf(TensorType type) {
             DimensionSizes.Builder b = new DimensionSizes.Builder(type.dimensions().size());
             for (int i = 0; i < type.dimensions().size(); i++)
                 b.set(i, type.dimensions().get(i).size().get());