Skip to content

Commit

Permalink
Fixed pyi file build when OpenVINO_DIR is externally defined
Browse files Browse the repository at this point in the history
  • Loading branch information
ilya-lavrenov committed Nov 28, 2024
1 parent 86068a5 commit ffe042b
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 1 deletion.
14 changes: 14 additions & 0 deletions src/python/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -114,6 +114,20 @@ elseif(DEFINED PY_BUILD_CMAKE_PACKAGE_NAME AND NOT WIN32)
# in case of wheel build, pybind11-stubgen is always available via pyproject.toml's build-system
# except Win32 where we have issues with pybind11_stubgen executable which cannot import its own module
set(pybind11_stubgen_AVAILABLE ON)

# by default, wheel build is performed with build-isolation, which means that some variables like PYTHONPATH
# are not available. But if user called setupvars.sh, then OpenVINO dir is available, while PYTHONPATH - no.
# In this case, we will have mismatch on Linux when OpenVINO can point on build dir / install dir, while
# PYTHONPATH points out to locally installed tmp OpenVINO wheel (build against wheel).
# Ways to handle it:
# - setting PYTHONPATH to $ENV{INTEL_OPENVINO_DIR}/python if INTEL_OPENVINO_DIR is defined. It means we are building against
# OpenVINO archive or installation tree
# - if it's not defined, we cannot do any guesses and hence, disable pybind11-stubgen usage
if(DEFINED ENV{INTEL_OPENVINO_DIR})
set(openvino_pythonpath "$ENV{INTEL_OPENVINO_DIR}/python")
elseif(LINUX AND NOT OpenVINO_DIR STREQUAL OpenVINO_DIR_PY)
set(pybind11_stubgen_AVAILABLE OFF)
endif()
endif()

# but we also need to check whether OpenVINO is installed
Expand Down
2 changes: 1 addition & 1 deletion src/python/py_vlm_pipeline.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ auto vlm_generate_docstring = R"(
:type images: List[ov.Tensor] or ov.Tensor
:param generation_config: generation_config
:type generation_config: GenerationConfig or a Dict
:type generation_config: GenerationConfig or a Dict
:param streamer: streamer either as a lambda with a boolean returning flag whether generation should be stopped
:type : Callable[[str], bool], ov.genai.StreamerBase
Expand Down

0 comments on commit ffe042b

Please sign in to comment.