Fix Pyright errors in utils/types.py

src/bindings/python/pyproject.toml

@@ -5,7 +5,6 @@ include = '\.pyi?$'
 [tool.pyright]
 exclude = [
     "src/openvino/utils/data_helpers/data_dispatcher.py",
-    "src/openvino/utils/types.py",
     "src/openvino/**/ops.py",
     "src/openvino/utils/decorators.py",
     "src/openvino/frontend/frontend.py",

src/bindings/python/src/openvino/_ov_api.pyi

@@ -519,18 +519,30 @@ class Model:
                             :type name: str
         """
     @typing.overload
-    def __init__(self: openvino._pyopenvino.Model, results: list[openvino._pyopenvino.Node], parameters: list[openvino._pyopenvino.op.Parameter], name: str = '') -> None:
+    def __init__(self: openvino._pyopenvino.Model, results: list[openvino._pyopenvino.op.Result], parameters: list[openvino._pyopenvino.op.Parameter], name: str = '') -> None:
         """
                             Create user-defined Model which is a representation of a model.
 
-                            :param results: List of Nodes to be used as results.
-                            :type results: List[openvino.Node]
+                            :param results: List of results.
+                            :type results: List[op.Result]
                             :param parameters: List of parameters.
-                            :type parameters:  List[op.Parameter]
+                            :type parameters: List[op.Parameter]
                             :param name: String to set as model's friendly name.
                             :type name: str
         """
     @typing.overload
+    def __init__(self: openvino._pyopenvino.Model, results: list[openvino._pyopenvino.Node], parameters: list[openvino._pyopenvino.op.Parameter], name: str = '') -> None:
+        """
+                    Create user-defined Model which is a representation of a model.
+
+                    :param results: List of Nodes to be used as results.
+                    :type results: List[openvino.Node]
+                    :param parameters: List of parameters.
+                    :type parameters:  List[op.Parameter]
+                    :param name: String to set as model's friendly name.
+                    :type name: str
+        """
+    @typing.overload
     def __init__(self: openvino._pyopenvino.Model, result: openvino._pyopenvino.Node, parameters: list[openvino._pyopenvino.op.Parameter], name: str = '') -> None:
         """
                             Create user-defined Model which is a representation of a model.

src/bindings/python/src/openvino/_pyopenvino/__init__.pyi

@@ -2277,18 +2277,30 @@ class Model:
                             :type name: str
         """
     @typing.overload
-    def __init__(self, results: list[Node], parameters: list[op.Parameter], name: str = '') -> None:
+    def __init__(self, results: list[op.Result], parameters: list[op.Parameter], name: str = '') -> None:
         """
                             Create user-defined Model which is a representation of a model.
 
-                            :param results: List of Nodes to be used as results.
-                            :type results: List[openvino.Node]
+                            :param results: List of results.
+                            :type results: List[op.Result]
                             :param parameters: List of parameters.
-                            :type parameters:  List[op.Parameter]
+                            :type parameters: List[op.Parameter]
                             :param name: String to set as model's friendly name.
                             :type name: str
         """
     @typing.overload
+    def __init__(self, results: list[Node], parameters: list[op.Parameter], name: str = '') -> None:
+        """
+                    Create user-defined Model which is a representation of a model.
+
+                    :param results: List of Nodes to be used as results.
+                    :type results: List[openvino.Node]
+                    :param parameters: List of parameters.
+                    :type parameters:  List[op.Parameter]
+                    :param name: String to set as model's friendly name.
+                    :type name: str
+        """
+    @typing.overload
     def __init__(self, result: Node, parameters: list[op.Parameter], name: str = '') -> None:
         """
                             Create user-defined Model which is a representation of a model.

src/bindings/python/src/openvino/_pyopenvino/op/__init__.pyi

@@ -12,6 +12,10 @@ class Constant(openvino._pyopenvino.Node):
     """
     openvino.op.Constant wraps ov::op::v0::Constant
     """
+    def __buffer__(self, flags):
+        """
+        Return a buffer object that exposes the underlying memory of the object.
+        """
     @typing.overload
     def __init__(self, array: numpy.ndarray[typing.Any, numpy.dtype[typing.Any]], shared_memory: bool = False) -> None:
         ...
@@ -54,6 +58,10 @@ class Constant(openvino._pyopenvino.Node):
     @typing.overload
     def __init__(self, arg0: openvino._pyopenvino.Type, arg1: openvino._pyopenvino.Shape, arg2: list[int]) -> None:
         ...
+    def __release_buffer__(self, buffer):
+        """
+        Release the buffer object that exposes the underlying memory of the object.
+        """
     def __repr__(self) -> str:
         ...
     def get_byte_size(self) -> int:

src/bindings/python/src/openvino/opset1/ops.pyi

@@ -1772,7 +1772,7 @@ def transpose(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: Transpose node
 
     """
-def unsqueeze(data: typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], axes: typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], name: typing.Optional[str] = None) -> openvino._pyopenvino.Node:
+def unsqueeze(data: typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray], axes: typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray], name: typing.Optional[str] = None) -> openvino._pyopenvino.Node:
     """
     Perform unsqueeze operation on input tensor.
 
@@ -1799,7 +1799,7 @@ def variadic_split(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: VariadicSplit node
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericData: typing._UnionGenericAlias  # value = typing.Union[int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 ScalarData: typing._UnionGenericAlias  # value = typing.Union[int, float]

src/bindings/python/src/openvino/opset10/ops.pyi

@@ -102,6 +102,6 @@ def unique(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: Node representing Unique operation.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 _get_node_factory_opset10: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset10')
 _get_node_factory_opset4: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset4')

src/bindings/python/src/openvino/opset11/ops.pyi

@@ -56,5 +56,5 @@ def topk(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node which performs TopK
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 _get_node_factory_opset11: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset11')

src/bindings/python/src/openvino/opset12/ops.pyi

@@ -64,5 +64,5 @@ def scatter_elements_update(*args, **kwargs) -> openvino._pyopenvino.Node:
         to `axis` is obtained from the index of the entry itself.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 _get_node_factory_opset12: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset12')

src/bindings/python/src/openvino/opset13/ops.pyi

@@ -188,7 +188,7 @@ def scaled_dot_product_attention(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node performing Scaled Dot Product Attention operation.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericData: typing._UnionGenericAlias  # value = typing.Union[int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 _get_node_factory_opset13: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset13')

src/bindings/python/src/openvino/opset14/ops.pyi

@@ -83,6 +83,6 @@ def max_pool(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return:   The new node performing max pooling operation.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 TensorShape: typing._GenericAlias  # value = typing.List[int]
 _get_node_factory_opset14: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset14')

src/bindings/python/src/openvino/opset15/ops.pyi

@@ -202,6 +202,6 @@ def string_tensor_unpack(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node performing StringTensorUnpack operation.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 _get_node_factory_opset15: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset15')
 constant: openvino.utils.decorators.MultiMethod  # value = <openvino.utils.decorators.MultiMethod object>

src/bindings/python/src/openvino/opset16/ops.pyi

@@ -49,5 +49,5 @@ def segment_max(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node performing SegmentMax operation.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 _get_node_factory_opset16: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset16')

src/bindings/python/src/openvino/opset2/ops.pyi

@@ -115,7 +115,7 @@ def space_to_batch(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node performing a SpaceToBatch operation.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericData: typing._UnionGenericAlias  # value = typing.Union[int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 ScalarData: typing._UnionGenericAlias  # value = typing.Union[int, float]

src/bindings/python/src/openvino/opset3/ops.pyi

@@ -356,7 +356,7 @@ def topk(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node which performs TopK (both indices and values)
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericData: typing._UnionGenericAlias  # value = typing.Union[int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 ScalarData: typing._UnionGenericAlias  # value = typing.Union[int, float]

src/bindings/python/src/openvino/opset4/ops.pyi

@@ -278,7 +278,7 @@ def swish(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node which performs Swish
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericData: typing._UnionGenericAlias  # value = typing.Union[int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 ScalarData: typing._UnionGenericAlias  # value = typing.Union[int, float]

src/bindings/python/src/openvino/opset5/ops.pyi

@@ -83,7 +83,7 @@ def gru_sequence(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node represents GRUSequence. Node outputs count: 2.
 
     """
-def hsigmoid(data: typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], name: typing.Optional[str] = None) -> openvino._pyopenvino.Node:
+def hsigmoid(data: typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray], name: typing.Optional[str] = None) -> openvino._pyopenvino.Node:
     """
     Return a node which performs HSigmoid.
 
@@ -186,7 +186,7 @@ def round(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node with Round operation applied on each element.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericData: typing._UnionGenericAlias  # value = typing.Union[int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 ScalarData: typing._UnionGenericAlias  # value = typing.Union[int, float]

src/bindings/python/src/openvino/opset6/ops.pyi

@@ -60,7 +60,7 @@ def mvn(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node performing a MVN operation on input tensor.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 TensorShape: typing._GenericAlias  # value = typing.List[int]
 _get_node_factory_opset6: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset6')

src/bindings/python/src/openvino/opset7/ops.pyi

@@ -28,7 +28,7 @@ import typing
 Factory functions for all openvino ops.
 """
 __all__ = ['Constant', 'Node', 'NodeFactory', 'NodeInput', 'NumericData', 'NumericType', 'Parameter', 'ScalarData', 'Shape', 'TensorShape', 'as_node', 'as_nodes', 'assert_list_of_ints', 'binary_op', 'check_valid_attributes', 'dft', 'einsum', 'gather', 'gelu', 'get_dtype', 'get_element_type', 'get_element_type_str', 'idft', 'is_non_negative_value', 'is_positive_value', 'make_constant_node', 'nameable_op', 'np', 'partial', 'roll', 'unary_op']
-def dft(data: typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], axes: typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], signal_size: typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray, NoneType] = None, name: typing.Optional[str] = None) -> openvino._pyopenvino.Node:
+def dft(data: typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray], axes: typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray], signal_size: typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray, NoneType] = None, name: typing.Optional[str] = None) -> openvino._pyopenvino.Node:
     """
     Return a node which performs DFT operation.
 
@@ -93,7 +93,7 @@ def roll(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node performing a Roll operation on the input tensor.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 NumericData: typing._UnionGenericAlias  # value = typing.Union[int, float, numpy.ndarray]
 NumericType: typing._UnionGenericAlias  # value = typing.Union[type, numpy.dtype]
 ScalarData: typing._UnionGenericAlias  # value = typing.Union[int, float]

src/bindings/python/src/openvino/opset8/ops.pyi

@@ -448,6 +448,6 @@ def softmax(*args, **kwargs) -> openvino._pyopenvino.Node:
         :return: The new node with softmax operation applied on each element.
 
     """
-NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, int, float, numpy.ndarray]
+NodeInput: typing._UnionGenericAlias  # value = typing.Union[openvino._pyopenvino.Node, openvino._pyopenvino.Output, int, float, numpy.ndarray]
 TensorShape: typing._GenericAlias  # value = typing.List[int]
 _get_node_factory_opset8: functools.partial  # value = functools.partial(<function _get_node_factory at memory_address>, 'opset8')