diff options
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.java | 2 |
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 |