diff options
-rw-r--r-- | pkgs/development/python-modules/lightgbm/default.nix | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/lightgbm/default.nix b/pkgs/development/python-modules/lightgbm/default.nix index 71604af7297..a313aa2233e 100644 --- a/pkgs/development/python-modules/lightgbm/default.nix +++ b/pkgs/development/python-modules/lightgbm/default.nix @@ -12,7 +12,7 @@ , ocl-icd , opencl-headers , boost -, gpuSupport ? true +, gpuSupport ? stdenv.isLinux }: buildPythonPackage rec { @@ -48,7 +48,11 @@ buildPythonPackage rec { ]; buildPhase = '' + runHook preBuild + ${python.pythonForBuild.interpreter} setup.py bdist_wheel ${lib.optionalString gpuSupport "--gpu"} + + runHook postBuild ''; postConfigure = '' |