diff options
author | Lester Solbakken <lesters@oath.com> | 2021-04-22 14:04:43 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-04-22 14:04:43 +0200 |
commit | a242591a9328fa21959ca76c08e616b1f1c682d7 (patch) | |
tree | 93d0035a9169fa17758fdcd5e52ec233fbd1cdd2 /vespajlib | |
parent | 3410aa89214638e462ecfa72ce3e93fe4aa38c7d (diff) |
Wire in tensor cell type resolving for rename in Java
Diffstat (limited to 'vespajlib')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java index 275b546c0aa..fc1e7737d83 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java @@ -5,6 +5,7 @@ import com.google.common.collect.ImmutableList; import com.yahoo.tensor.Tensor; import com.yahoo.tensor.TensorAddress; import com.yahoo.tensor.TensorType; +import com.yahoo.tensor.TypeResolver; import com.yahoo.tensor.evaluation.EvaluationContext; import com.yahoo.tensor.evaluation.Name; import com.yahoo.tensor.evaluation.TypeContext; @@ -76,10 +77,7 @@ public class Rename<NAMETYPE extends Name> extends PrimitiveTensorFunction<NAMET } private TensorType type(TensorType type) { - TensorType.Builder builder = new TensorType.Builder(type.valueType()); - for (TensorType.Dimension dimension : type.dimensions()) - builder.dimension(dimension.withName(fromToMap.getOrDefault(dimension.name(), dimension.name()))); - return builder.build(); + return TypeResolver.rename(type, fromDimensions, toDimensions); } @Override |