Use CONDA_PREFIX/include for zlib.h in a conda environment

BoPeng · Oct 10, 2018 · 5e44e87 · 5e44e87
1 parent 8c23724
commit 5e44e87
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -579,9 +579,9 @@ def is_maverick():
         else:
             common_extra_compile_args.append('/openmp')
 else:
-    if os.path.isdir('/opt/conda'):
-        common_extra_include_dirs.append('/opt/conda/include')
-        common_library_dirs.append('/opt/conda/lib')
+    if 'CONDA_PREFIX' in os.environ:
+        common_extra_include_dirs.append(os.environ['CONDA_PREFIX'] + '/include')
+        common_library_dirs.append(os.environ['CONDA_PREFIX'] + 'lib')
     common_extra_compile_args = ['-O3', '-Wall', '-Wno-unknown-pragmas', '-Wno-unused-parameter']
     if is_maverick():
         common_extra_link_args.append('-stdlib=libstdc++')