force conda

Author: Tieqiong

setup.py

@@ -15,17 +15,25 @@
 import numpy
 from setuptools import Extension, setup
 
-
 def get_boost_libraries():
+    # Force the use of the conda-installed Boost.Python library.
+    conda_prefix = os.environ.get("CONDA_PREFIX")
+    if not conda_prefix:
+        raise EnvironmentError("CONDA_PREFIX is not set. Please activate your conda environment.")
+    if os.name == "nt":
+        lib_dir = Path(conda_prefix) / "Library" / "lib"
+    else:
+        lib_dir = Path(conda_prefix) / "lib"
     base_lib = "boost_python"
     major, minor = str(sys.version_info[0]), str(sys.version_info[1])
     tags = [f"{major}{minor}", major, ""]
     mttags = ["", "-mt"]
-    candidates = [base_lib + tag for tag in tags for mt in mttags] + [base_lib]
+    candidates = [base_lib + tag + mt for tag in tags for mt in mttags] + [base_lib]
     for lib in candidates:
-        if find_library(lib):
+        # Check for common shared library extensions.
+        if any((lib_dir / (lib + ext)).exists() for ext in [".so", ".dylib", ".dll", ".lib"]):
             return [lib]
-    raise RuntimeError("Cannot find a suitable Boost.Python library.")
+    raise RuntimeError("Cannot find a suitable Boost.Python library in your conda environment.")
 
 
 def get_boost_config():