Merge pull request #124316 from risicle/ris-lightgbm-darwin

python3Packages.lightgbm: fix for darwin
This commit is contained in:
Robert Scott 2021-05-25 19:43:12 +01:00 committed by GitHub
commit c29e090a7d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -23,19 +23,7 @@ buildPythonPackage rec {
dontUseCmakeConfigure = true;
# we never actually explicitly call the install command so this is the only way
# to inject these options to it - however, openmp-library doesn't appear to have
# any effect, so we have to inject it into NIX_LDFLAGS manually below
postPatch = lib.optionalString stdenv.cc.isClang ''
cat >> setup.cfg <<EOF
[install]
openmp-include-dir=${llvmPackages.openmp}/include
openmp-library=${llvmPackages.openmp}/lib/libomp.dylib
EOF
'';
buildInputs = lib.optional stdenv.cc.isClang [ llvmPackages.openmp ];
propagatedBuildInputs = [
numpy
scipy
@ -44,14 +32,13 @@ buildPythonPackage rec {
postConfigure = ''
export HOME=$(mktemp -d)
'' + lib.optionalString stdenv.cc.isClang ''
export NIX_LDFLAGS="$NIX_LDFLAGS -L${llvmPackages.openmp}/lib -lomp"
'';
# The pypi package doesn't distribute the tests from the GitHub
# repository. It contains c++ tests which don't seem to wired up to
# `make check`.
doCheck = false;
pythonImportsCheck = [ "lightgbm" ];
meta = with lib; {
description = "A fast, distributed, high performance gradient boosting (GBDT, GBRT, GBM or MART) framework";