do not require any specific Value subclass

1 files changed, 1 insertions, 4 deletions

diff --git a/searchlib/src/main/java/com/yahoo/searchlib/tensor/EvaluateTensorConformance.java b/searchlib/src/main/java/com/yahoo/searchlib/tensor/EvaluateTensorConformance.java
index cf506efcb78..bd6fb47be1e 100644
--- a/searchlib/src/main/java/com/yahoo/searchlib/tensor/EvaluateTensorConformance.java
+++ b/searchlib/src/main/java/com/yahoo/searchlib/tensor/EvaluateTensorConformance.java
@@ -103,10 +103,7 @@ public class EvaluateTensorConformance {
 
     private Tensor evaluate(String expression, MapContext context) throws ParseException {
         Value value = new RankingExpression(expression).evaluate(context);
-        if (!(value instanceof TensorValue)) {
-            throw new IllegalArgumentException("Result is not a tensor");
-        }
-        return ((TensorValue)value).asTensor();
+        return value.asTensor();
     }
 
     private MapContext getInput(Inspector inputs) {