diff --git a/modules/nvidia_plugin/tests/unit/result.cpp b/modules/nvidia_plugin/tests/unit/result.cpp index 0660e606f..105af4beb 100644 --- a/modules/nvidia_plugin/tests/unit/result.cpp +++ b/modules/nvidia_plugin/tests/unit/result.cpp @@ -25,7 +25,9 @@ class ov::Output : public ov::Output { public: explicit Output(std::shared_ptr node) : ov::Output(node, 0) { auto tensor = std::make_shared( - ov::element::Type{}, ov::PartialShape{1}, ParameterStubNode::get_type_info_static().name); + ov::element::Type{}, + ov::PartialShape{1}, + std::unordered_set{ParameterStubNode::get_type_info_static().name}); node->m_outputs.emplace_back(node.get(), 0, tensor); } };