1 files changed, 1 insertions, 6 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java b/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
index 107925de8c0..42df2877537 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/Tensor.java
@@ -2,7 +2,6 @@
 package com.yahoo.tensor;
 
 import com.google.common.annotations.Beta;
-import com.google.common.collect.ImmutableMap;
 import com.yahoo.tensor.functions.ConstantTensor;
 import com.yahoo.tensor.functions.Generate;
 import com.yahoo.tensor.functions.Join;
@@ -16,16 +15,12 @@ import com.yahoo.tensor.functions.Softmax;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
-import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
-import java.util.Set;
 import java.util.function.DoubleBinaryOperator;
 import java.util.function.DoubleUnaryOperator;
 import java.util.function.Function;
 
-import static com.yahoo.tensor.TensorType.Dimension.Type.mapped;
-
 /**
  * A multidimensional array which can be used in computations.
  * <p>
@@ -229,7 +224,7 @@ public interface Tensor {
         if (containsIndexedDimensions && containsMappedDimensions)
             throw new IllegalArgumentException("Mixed dimension types are not supported, got: " + type);
         if (containsMappedDimensions)
-            return MapTensor.from(type, tensorString);
+            return MappedTensor.from(type, tensorString);
         else // indexed or none
             return IndexedTensor.from(type, tensorString);
     }