summaryrefslogtreecommitdiffstats
path: root/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java
diff options
context:
space:
mode:
Diffstat (limited to 'model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java')
-rw-r--r--model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java2
1 files changed, 1 insertions, 1 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java
index 4a1ee22d288..a7b536ba911 100644
--- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java
+++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/LazyValue.java
@@ -42,7 +42,7 @@ class LazyValue extends Value {
@Override
public TensorType type() {
- return computedValue().type(); // TODO: Keep type information in this/ExpressionFunction to avoid computing here
+ return model.requireReferencedFunction(function).returnType().get();
}
@Override