Add support for additional CMake arguments

quantumlib · Jan 30, 2024 · 5ea50be · 5ea50be
1 parent 11523c5
commit 5ea50be
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -48,6 +48,10 @@ def build_extension(self, ext):
             "-DPYTHON_INCLUDE_DIR=" + python_include_dir,
         ]
 
+        additional_cmake_args = os.environ.get("CMAKE_ARGS", "")
+        if additional_cmake_args:
+            cmake_args += additional_cmake_args.split()
+
         cfg = "Debug" if self.debug else "Release"
         build_args = ["--config", cfg]