IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[PyOV] fix set_node_friendly_name (#20695)

Browse Source
This commit is contained in:
Anastasia Kuporosova 2023-10-26 15:35:57 +03:00 committed by GitHub
parent c482e5f807
commit 3b9606f217
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/bindings/python/src/openvino/runtime/utils/decorators.py
View File

@ -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

6
src/bindings/python/tests/test_graph/test_reduction.py
View File

@ -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