Add classes to ov.__init__

Signed-off-by: Alicja Miloszewska <[email protected]>
openvinotoolkit · Nov 12, 2024 · e0c86e9 · e0c86e9
1 parent 55d8038
commit e0c86e9
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 5 deletions.
diff --git a/src/bindings/python/src/openvino/__init__.py b/src/bindings/python/src/openvino/__init__.py
@@ -34,6 +34,9 @@
 
 from openvino._pyopenvino import Symbol
 from openvino._pyopenvino import Dimension
+from openvino._pyopenvino import Input
+from openvino._pyopenvino import Output
+from openvino._pyopenvino import Node
 from openvino._pyopenvino import Strides
 from openvino._pyopenvino import PartialShape
 from openvino._pyopenvino import Shape

diff --git a/src/bindings/python/src/openvino/frontend/jax/jaxpr_decoder.py b/src/bindings/python/src/openvino/frontend/jax/jaxpr_decoder.py
@@ -6,7 +6,7 @@
 
 import jax.core
 from openvino.frontend.jax.py_jax_frontend import _FrontEndJaxDecoder as Decoder
-from openvino._pyopenvino import PartialShape, Type as OVType, OVAny
+from openvino import PartialShape, Type as OVType, OVAny
 from openvino.frontend.jax.utils import jax_array_to_ov_const, get_ov_type_for_value, \
     ivalue_to_constant, param_to_constants
 

diff --git a/src/bindings/python/src/openvino/frontend/pytorch/fx_decoder.py b/src/bindings/python/src/openvino/frontend/pytorch/fx_decoder.py
@@ -10,7 +10,7 @@
 
 from openvino.frontend.pytorch.py_pytorch_frontend import _FrontEndPytorchDecoder as Decoder
 from openvino.frontend.pytorch.py_pytorch_frontend import _Type as DecoderType
-from openvino._pyopenvino import PartialShape, Type as OVType, OVAny, Shape
+from openvino import PartialShape, Type as OVType, OVAny, Shape
 from openvino.frontend.pytorch.utils import make_constant, fetch_attr, pt_to_ov_type_map, torch_tensor_to_ov_const
 
 logger = logging.getLogger(__name__)

diff --git a/src/bindings/python/src/openvino/opset_utils.py b/src/bindings/python/src/openvino/opset_utils.py
@@ -5,7 +5,7 @@
 from typing import Optional
 import numpy as np
 
-from openvino._pyopenvino import Node
+from openvino import Node
 from openvino.utils.decorators import nameable_op
 from openvino.utils.node_factory import NodeFactory
 from openvino.utils.types import (

diff --git a/tools/benchmark_tool/openvino/__init__.py b/tools/benchmark_tool/openvino/__init__.py
@@ -18,7 +18,6 @@
 # # It is not compared with init files from openvino-dev package.
 # #
 # Import all public modules
-# from openvino import runtime as runtime
 from openvino import frontend as frontend
 from openvino import helpers as helpers
 from openvino import experimental as experimental
@@ -35,6 +34,9 @@
 
 from openvino._pyopenvino import Symbol
 from openvino._pyopenvino import Dimension
+from openvino._pyopenvino import Input
+from openvino._pyopenvino import Output
+from openvino._pyopenvino import Node
 from openvino._pyopenvino import Strides
 from openvino._pyopenvino import PartialShape
 from openvino._pyopenvino import Shape

diff --git a/tools/ovc/openvino/__init__.py b/tools/ovc/openvino/__init__.py
@@ -18,7 +18,6 @@
 # # It is not compared with init files from openvino-dev package.
 # #
 # Import all public modules
-# from openvino import runtime as runtime
 from openvino import frontend as frontend
 from openvino import helpers as helpers
 from openvino import experimental as experimental
@@ -35,6 +34,9 @@
 
 from openvino._pyopenvino import Symbol
 from openvino._pyopenvino import Dimension
+from openvino._pyopenvino import Input
+from openvino._pyopenvino import Output
+from openvino._pyopenvino import Node
 from openvino._pyopenvino import Strides
 from openvino._pyopenvino import PartialShape
 from openvino._pyopenvino import Shape