aboutsummaryrefslogtreecommitdiffstats
path: root/eval/src/tests/tensor/tensor_mapper/tensor_mapper_test.cpp
diff options
context:
space:
mode:
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.cpp5
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);