diff --git a/pkgs/development/python-modules/lightgbm/default.nix b/pkgs/development/python-modules/lightgbm/default.nix index 71604af72977..a313aa2233ec 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 = ''