1 files changed, 7 insertions, 0 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/query/ranking/RankFeatures.java b/container-search/src/main/java/com/yahoo/search/query/ranking/RankFeatures.java
index 1a4ecb4ecd8..807d70739cc 100644
--- a/container-search/src/main/java/com/yahoo/search/query/ranking/RankFeatures.java
+++ b/container-search/src/main/java/com/yahoo/search/query/ranking/RankFeatures.java
@@ -2,8 +2,10 @@
 package com.yahoo.search.query.ranking;
 
 import com.yahoo.fs4.MapEncoder;
+import com.yahoo.processing.request.CompoundName;
 import com.yahoo.search.Query;
 import com.yahoo.search.query.Ranking;
+import com.yahoo.search.query.profile.types.QueryProfileType;
 import com.yahoo.tensor.Tensor;
 import com.yahoo.text.JSON;
 
@@ -47,9 +49,14 @@ public class RankFeatures implements Cloneable {
 
     /** Sets a tensor rank feature */
     public void put(String name, Tensor value) {
+        verifyType(name, value);
         features.put(name, value);
     }
 
+    private void verifyType(String name, Object value) {
+        parent.getParent().properties().requireSettable(new CompoundName(List.of("ranking", "features", name)), value, Map.of());
+    }
+
     /**
      * Sets a rank feature to a value represented as a string.
      *