Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.3 kB · 122 lines
Nix
at flake-libs
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123{ lib, config, stdenv, pkgs, buildPythonPackage, fetchPypi,
# build-system scikit-build-core,
# nativeBuildInputs cmake, ninja, pathspec, pyproject-metadata, writableTmpDirAsHomeHook,
# buildInputs llvmPackages, boost, ocl-icd, opencl-headers,
# dependencies numpy, scipy,
# optional-dependencies cffi, dask, pandas, pyarrow, scikit-learn,
# optionals: gpu gpuSupport ? stdenv.hostPlatform.isLinux && !cudaSupport, cudaSupport ? config.cudaSupport, cudaPackages,}:
assert gpuSupport -> !cudaSupport;assert cudaSupport -> !gpuSupport;
buildPythonPackage rec { inherit (pkgs.lightgbm) pname version patches ; pyproject = true;
src = fetchPypi { inherit pname version; hash = "sha256-yxxZcg61aTicC6dNFPUjUbVzr0ifIwAyocnzFPi6t/4="; };
build-system = [ scikit-build-core ];
nativeBuildInputs = [ cmake ninja pathspec pyproject-metadata writableTmpDirAsHomeHook ] ++ lib.optionals cudaSupport [ cudaPackages.cuda_nvcc ];
dontUseCmakeConfigure = true;
buildInputs = (lib.optionals stdenv.cc.isClang [ llvmPackages.openmp ]) ++ (lib.optionals gpuSupport [ boost ocl-icd opencl-headers ]) ++ lib.optionals cudaSupport [ cudaPackages.cuda_nvcc cudaPackages.cuda_cudart ];
dependencies = [ numpy scipy ];
pypaBuildFlags = lib.optionals gpuSupport [ "--config-setting=cmake.define.USE_GPU=ON" ] ++ lib.optionals cudaSupport [ "--config-setting=cmake.define.USE_CUDA=ON" ];
optional-dependencies = { arrow = [ cffi pyarrow ]; dask = [ dask pandas ] ++ dask.optional-dependencies.array ++ dask.optional-dependencies.dataframe ++ dask.optional-dependencies.distributed; pandas = [ pandas ]; scikit-learn = [ scikit-learn ]; };
# No python tests doCheck = false;
pythonImportsCheck = [ "lightgbm" ];
meta = { description = "Fast, distributed, high performance gradient boosting (GBDT, GBRT, GBM or MART) framework"; homepage = "https://github.com/Microsoft/LightGBM"; changelog = "https://github.com/microsoft/LightGBM/releases/tag/v${version}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ teh ]; };}