diff options
Diffstat (limited to 'eval')
-rw-r--r-- | eval/src/vespa/eval/eval/tensor_spec.h | 1 | ||||
-rw-r--r-- | eval/src/vespa/eval/eval/test/eval_fixture.h | 1 |
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; } |