diff --git a/src/core/tests/preprocess.cpp b/src/core/tests/preprocess.cpp index 4ecb889b943..9c472af835f 100644 --- a/src/core/tests/preprocess.cpp +++ b/src/core/tests/preprocess.cpp @@ -1178,25 +1178,25 @@ TEST(pre_post_process, postprocess_convert_layout_invalid_dims_dyn_shape) { TEST(pre_post_process, postprocess_preserve_rt_info) { auto f = create_simple_function(element::f32, Shape{1, 3, 2, 2}); - f->get_results()[0]->get_rt_info()["someKey"] = ov::make_variant("someValue"); - f->input().get_rt_info()["someKey_in"] = ov::make_variant("someValue_in"); - f->output().get_rt_info()["someKey_out"] = ov::make_variant("someValue_out"); + f->get_results()[0]->get_rt_info()["someKey"] = "someValue"; + f->input().get_rt_info()["someKey_in"] = "someValue_in"; + f->output().get_rt_info()["someKey_out"] = "someValue_out"; auto p = PrePostProcessor(f); p.output().tensor().set_element_type(element::u8); f = p.build(); EXPECT_EQ(f->output().get_element_type(), element::u8); ASSERT_EQ(f->get_results()[0]->get_rt_info().count("someKey"), 1); - auto var0 = std::dynamic_pointer_cast>(f->get_results()[0]->get_rt_info()["someKey"]); - EXPECT_EQ(var0->get(), "someValue"); + auto var0 = f->get_results()[0]->get_rt_info()["someKey"].as(); + EXPECT_EQ(var0, "someValue"); ASSERT_EQ(f->input().get_rt_info().count("someKey_in"), 1); - auto var0_in = std::dynamic_pointer_cast>(f->input().get_rt_info()["someKey_in"]); - EXPECT_EQ(var0_in->get(), "someValue_in"); + auto var0_in = f->input().get_rt_info()["someKey_in"].as(); + EXPECT_EQ(var0_in, "someValue_in"); ASSERT_EQ(f->output().get_rt_info().count("someKey_out"), 1); - auto var0_out = std::dynamic_pointer_cast>(f->output().get_rt_info()["someKey_out"]); - EXPECT_EQ(var0_out->get(), "someValue_out"); + auto var0_out = f->output().get_rt_info()["someKey_out"].as(); + EXPECT_EQ(var0_out, "someValue_out"); } TEST(pre_post_process, postprocess_custom_step) {