nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-26 05:57:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

python3Packages.scikit-learn: enable parallel building and testing

Browse Source
This commit is contained in:
DavHau 2021-05-15 15:14:01 +07:00
parent cd8f3e6c44
commit 0a1a14d13e
1 changed files with 21 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
pkgs/development/python-modules/scikit-learn/default.nix
View File

@ -7,7 +7,8 @@
, glibcLocales , glibcLocales
, numpy , numpy
, scipy , scipy
, pytest , pytestCheckHook
, pytest-xdist
, pillow , pillow
, cython , cython
, joblib , joblib
@ -54,17 +55,30 @@ buildPythonPackage rec {
joblib joblib
threadpoolctl threadpoolctl
]; ];
checkInputs = [ pytest ];
checkInputs = [ pytestCheckHook pytest-xdist ];
LC_ALL="en_US.UTF-8"; LC_ALL="en_US.UTF-8";
doCheck = !stdenv.isAarch64; preBuild = ''
# Skip test_feature_importance_regression - does web fetch export SKLEARN_BUILD_PARALLEL=$NIX_BUILD_CORES
checkPhase = ''
cd $TMPDIR
HOME=$TMPDIR OMP_NUM_THREADS=1 pytest -k "not test_feature_importance_regression" --pyargs sklearn
''; '';
doCheck = !stdenv.isAarch64;
# Skip test_feature_importance_regression - does web fetch
disabledTests = [ "test_feature_importance_regression" ];
pytestFlagsArray = [ "-n" "$NIX_BUILD_CORES" "--pyargs" "sklearn" ];
preCheck = ''
cd $TMPDIR
export HOME=$TMPDIR
export OMP_NUM_THREADS=1
'';
pythonImportsCheck = [ "sklearn" ];
meta = with lib; { meta = with lib; {
description = "A set of python modules for machine learning and data mining"; description = "A set of python modules for machine learning and data mining";
changelog = let changelog = let