Add cmake option to anable/disable bindings

openvinotoolkit · Jan 10, 2025 · ad44de2 · ad44de2
1 parent f9019e0
commit ad44de2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/model_api/cpp/CMakeLists.txt b/model_api/cpp/CMakeLists.txt
@@ -4,6 +4,8 @@
 
 cmake_minimum_required(VERSION 3.26)
 
+option(ENABLE_PY_BINDINGS "Enables building python bindings package" ON)
+
 # Multi config generators such as Visual Studio ignore CMAKE_BUILD_TYPE. Multi config generators are configured with
 # CMAKE_CONFIGURATION_TYPES, but limiting options in it completely removes such build options
 get_property(GENERATOR_IS_MULTI_CONFIG_VAR GLOBAL PROPERTY GENERATOR_IS_MULTI_CONFIG)
@@ -84,7 +86,9 @@ elseif(CMAKE_CXX_COMPILER_ID MATCHES "^GNU|(Apple)?Clang$")
     set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC")
 endif()
 
-add_subdirectory(py_bindings)
+if (ENABLE_PY_BINDINGS)
+    add_subdirectory(py_bindings)
+endif()
 
 include(GenerateExportHeader)
 

diff --git a/model_api/cpp/py_bindings/py_classificaiton.cpp b/model_api/cpp/py_bindings/py_classificaiton.cpp
@@ -1,4 +1,3 @@
-
 /*
  * Copyright (C) 2025 Intel Corporation
  * SPDX-License-Identifier: Apache-2.0
@@ -54,7 +53,6 @@ void init_classification(nb::module_& m) {
             },
             nb::rv_policy::reference_internal);
 
-
     nb::class_<ClassificationModel, ImageModel>(m, "ClassificationModel")
         .def_static(
             "create_model",