[PyOV] fix set_node_friendly_name (#20695)

This commit is contained in:
Anastasia Kuporosova authored and GitHub committed 2023-10-26 14:35:57 +02:00
1 parent c482e5f807
commit 3b9606f217
2 files changed
+7 -1

No files matched your search

@@ -9,7 +9,7 @@ from openvino.runtime import Node, Output
from openvino.runtime.utils.types import NodeInput, as_node, as_nodes
def _set_node_friendly_name(node: Node, **kwargs: Any) -> Node:
def _set_node_friendly_name(node: Node, /, **kwargs: Any) -> Node:
if "name" in kwargs:
node.friendly_name = kwargs["name"]
return node
@@ -154,3 +154,9 @@ def test_normalize_l2():
assert node.get_output_size() == 1
assert node.get_type_name() == "NormalizeL2"
assert list(node.get_output_shape(0)) == input_shape
def test_reduce_with_keywork():
const = ov.constant([-1], np.int64)
min_op = ov.reduce_min(node=const, reduction_axes=0)
assert min_op.get_output_size() == 1