Update

[ghstack-poisoned]

Author: Vincent Moens

setup.py

@@ -129,9 +129,23 @@ def build_extension(self, ext):
             f"-DPython3_EXECUTABLE={get_python_executable()}",
         ]
         CONDA_PREFIX = os.environ.get("CONDA_PREFIX")
+        # if CONDA_PREFIX:
+        #     CMAKE_PREFIX_PATH = os.environ.get("CMAKE_PREFIX_PATH")
+        #     if CMAKE_PREFIX_PATH:
+        #         cmake_args.append(f"-DCMAKE_PREFIX_PATH={CONDA_PREFIX}:{CMAKE_PREFIX_PATH}")
+        #     else:
+        #         cmake_args.append(f"-DCMAKE_PREFIX_PATH={CONDA_PREFIX}")
         if CONDA_PREFIX:
-            CMAKE_PREFIX_PATH = os.environ.get("CMAKE_PREFIX_PATH")
-            cmake_args.append(f"-DCMAKE_PREFIX_PATH={CONDA_PREFIX}:{CMAKE_PREFIX_PATH}")
+            # Find pybind11
+            pybind11_dir = None
+            for config_file in ["pybind11Config.cmake", "pybind11-config.cmake"]:
+                config_path = glob.glob(os.path.join(CONDA_PREFIX, "**", config_file), recursive=True)
+                if config_path:
+                    pybind11_dir = os.path.dirname(config_path[0])
+                    break
+            if pybind11_dir:
+                cmake_args.append(f"-DPYBIND11_DIR={pybind11_dir}")
+
         build_args = []
         if not os.path.exists(self.build_temp):
             os.makedirs(self.build_temp)