make sure tensor spec has appropriate type

adds runtime support for input tensors with float cells in eval
fixture used for testing.

2 files changed, 2 insertions, 0 deletions

diff --git a/eval/src/vespa/eval/eval/tensor_spec.h b/eval/src/vespa/eval/eval/tensor_spec.h
index 25af4c7a93c..32dc1c82fcb 100644
--- a/eval/src/vespa/eval/eval/tensor_spec.h
+++ b/eval/src/vespa/eval/eval/tensor_spec.h
@@ -73,6 +73,7 @@ public:
         }
         return *this;
     }
+    void override_type(const vespalib::string &new_type) { _type = new_type; }
     const vespalib::string &type() const { return _type; }
     const Cells &cells() const { return _cells; }
     vespalib::string to_string() const;
diff --git a/eval/src/vespa/eval/eval/test/eval_fixture.h b/eval/src/vespa/eval/eval/test/eval_fixture.h
index e20d435f608..8c7d15e7416 100644
--- a/eval/src/vespa/eval/eval/test/eval_fixture.h
+++ b/eval/src/vespa/eval/eval/test/eval_fixture.h
@@ -29,6 +29,7 @@ public:
         std::map<vespalib::string,Param> map;
         ParamRepo() : map() {}
         ParamRepo &add(const vespalib::string &name, TensorSpec value_in, const vespalib::string &type_in, bool is_mutable_in) {
+            value_in.override_type(type_in);
             map.insert_or_assign(name, Param(std::move(value_in), type_in, is_mutable_in));
             return *this;
         }