diff options
Diffstat (limited to 'eval/src/tests/tensor/tensor_mapper/tensor_mapper_test.cpp')
-rw-r--r-- | eval/src/tests/tensor/tensor_mapper/tensor_mapper_test.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/eval/src/tests/tensor/tensor_mapper/tensor_mapper_test.cpp b/eval/src/tests/tensor/tensor_mapper/tensor_mapper_test.cpp index c26429f47e4..e369e09b99a 100644 --- a/eval/src/tests/tensor/tensor_mapper/tensor_mapper_test.cpp +++ b/eval/src/tests/tensor/tensor_mapper/tensor_mapper_test.cpp @@ -9,6 +9,7 @@ #include <vespa/eval/eval/simple_tensor.h> using vespalib::eval::ValueType; +using vespalib::eval::Value; using vespalib::eval::TensorSpec; using vespalib::eval::SimpleTensor; using namespace vespalib::tensor; @@ -21,8 +22,8 @@ void verify_wrapped(const TensorSpec &source, const vespalib::string &type, cons } void verify(const TensorSpec &source, const vespalib::string &type, const TensorSpec &expect) { - auto tensor = DefaultTensorEngine::ref().create(source); - const Tensor *tensor_impl = dynamic_cast<const Tensor *>(tensor.get()); + Value::UP value = DefaultTensorEngine::ref().from_spec(source); + const Tensor *tensor_impl = dynamic_cast<const Tensor *>(value->as_tensor()); ASSERT_TRUE(tensor_impl); TensorMapper mapper(ValueType::from_spec(type)); auto mapped = mapper.map(*tensor_impl); |