diff options
author | Arne Juul <arnej@verizonmedia.com> | 2021-04-23 12:31:21 +0000 |
---|---|---|
committer | Arne Juul <arnej@verizonmedia.com> | 2021-04-26 06:45:23 +0000 |
commit | fb98651ec8775d2347a3e42310e14d1b59c38a42 (patch) | |
tree | 52549d0bed084386fbbb015cb0278cda5db019ed /vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java | |
parent | c0b637dd81754665a014eba3794f31f7fd432d52 (diff) |
Reapply "Lesters/resolve cell types for rename and slice"
This reverts commit 49d9016a9e50fd9032b10520c71dc5b05ab8b215.
Diffstat (limited to 'vespajlib/src/main/java/com/yahoo/tensor/functions/Rename.java')
-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 |