Make stub generation optional

Simple-Robotics · Aug 8, 2024 · 609c415 · 609c415
1 parent eacf0ea
commit 609c415
Showing 1 changed file with 26 additions and 20 deletions.
diff --git a/bindings/python/CMakeLists.txt b/bindings/python/CMakeLists.txt
@@ -1,6 +1,8 @@
 include(${JRL_CMAKE_MODULES}/python.cmake)
 include(${JRL_CMAKE_MODULES}/python-helpers.cmake)
 
+option(GENERATE_PYTHON_STUBS "Generate Python stubs" OFF)
+
 findpython(REQUIRED Development.Module)
 set(Python_INCLUDE_DIRS ${PYTHON_INCLUDE_DIRS})
 
@@ -46,16 +48,18 @@ if(${CMAKE_SYSTEM_PROCESSOR} MATCHES "(x86)|(X86)|(amd64)|(AMD64)")
     set_target_properties(instructionset PROPERTIES INSTALL_RPATH
                                                     "\$ORIGIN/../../..")
   endif()
-  nanobind_add_stub(
-    instructionset_stub
-    MODULE
-    instructionset
-    OUTPUT
-    instructionset.pyi
-    PYTHON_PATH
-    $<TARGET_FILE_DIR:instructionset>
-    DEPENDS
-    instructionset)
+  if(GENERATE_PYTHON_STUBS)
+    nanobind_add_stub(
+      instructionset_stub
+      MODULE
+      instructionset
+      OUTPUT
+      instructionset.pyi
+      PYTHON_PATH
+      $<TARGET_FILE_DIR:instructionset>
+      DEPENDS
+      instructionset)
+  endif()
   install(
     TARGETS instructionset
     EXPORT ${TARGETS_EXPORT_NAME}
@@ -127,16 +131,18 @@ function(CREATE_PYTHON_TARGET target_name COMPILE_OPTIONS dependencies)
                                                     "\$ORIGIN/../../..")
   endif()
 
-  nanobind_add_stub(
-    ${target_name}_stub
-    MODULE
-    ${target_name}
-    OUTPUT
-    ${target_name}.pyi
-    PYTHON_PATH
-    $<TARGET_FILE_DIR:${target_name}>
-    DEPENDS
-    ${target_name})
+  if(GENERATE_PYTHON_STUBS)
+    nanobind_add_stub(
+      ${target_name}_stub
+      MODULE
+      ${target_name}
+      OUTPUT
+      ${target_name}.pyi
+      PYTHON_PATH
+      $<TARGET_FILE_DIR:${target_name}>
+      DEPENDS
+      ${target_name})
+  endif()
 
   install(TARGETS ${target_name} DESTINATION ${${PYWRAP}_INSTALL_DIR})
   install(FILES ${CMAKE_CURRENT_BINARY_DIR}/${target_name}.pyi