[one_hot]: Fixed tests.

pkowalc1 · pkowalc1 · commit f523520b7c7f · 2025-09-03T12:12:20.000Z
diff --git a/src/plugins/intel_cpu/src/nodes/one_hot.cpp b/src/plugins/intel_cpu/src/nodes/one_hot.cpp
@@ -64,7 +64,7 @@ OneHot::OneHot(const std::shared_ptr<ov::Node>& op, const GraphContext::CPtr& co
         OPENVINO_THROW_NOT_IMPLEMENTED(errorMessage);
     }
 
-    const auto oneHot = ov::as_type_ptr<const ov::op::v1::OneHot>(op);
+    const auto oneHot = ov::as_type_ptr<const ov::op::util::OneHotBase>(op);
     const auto depthNode = ov::as_type_ptr<const ov::op::v0::Constant>(oneHot->get_input_node_shared_ptr(DEPTH_ID));
     if (depthNode) {
         depth = depthNode->cast_vector<uint32_t>()[0];

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ OneHot::OneHot(const std::shared_ptr<ov::Node>& op, const GraphContext::CPtr& co`
`64`	`64`	`OPENVINO_THROW_NOT_IMPLEMENTED(errorMessage);`
`65`	`65`	`}`
`66`	`66`
`67`		`- const auto oneHot = ov::as_type_ptr<const ov::op::v1::OneHot>(op);`
	`67`	`+ const auto oneHot = ov::as_type_ptr<const ov::op::util::OneHotBase>(op);`
`68`	`68`	`const auto depthNode = ov::as_type_ptr<const ov::op::v0::Constant>(oneHot->get_input_node_shared_ptr(DEPTH_ID));`
`69`	`69`	`if (depthNode) {`
`70`	`70`	`depth = depthNode->cast_vector<uint32_t>()[0];`