parent
fd66e543ae
commit
2a9ea69cd5
|
@ -1,34 +1,31 @@
|
||||||
{ stdenv, buildPythonPackage, fetchPypi, python
|
{ stdenv, buildPythonPackage, fetchPypi, python
|
||||||
, nose, pillow
|
|
||||||
, gfortran, glibcLocales
|
, gfortran, glibcLocales
|
||||||
, numpy, scipy
|
, numpy, scipy, pytest, pillow
|
||||||
}:
|
}:
|
||||||
|
|
||||||
buildPythonPackage rec {
|
buildPythonPackage rec {
|
||||||
pname = "scikit-learn";
|
pname = "scikit-learn";
|
||||||
version = "0.19.2";
|
version = "0.20.0";
|
||||||
# UnboundLocalError: local variable 'message' referenced before assignment
|
# UnboundLocalError: local variable 'message' referenced before assignment
|
||||||
doCheck = false;
|
|
||||||
disabled = stdenv.isi686; # https://github.com/scikit-learn/scikit-learn/issues/5534
|
disabled = stdenv.isi686; # https://github.com/scikit-learn/scikit-learn/issues/5534
|
||||||
|
|
||||||
src = fetchPypi {
|
src = fetchPypi {
|
||||||
inherit pname version;
|
inherit pname version;
|
||||||
sha256 = "b276739a5f863ccacb61999a3067d0895ee291c95502929b2ae56ea1f882e888";
|
sha256 = "064cbxsis6m7l6pr09ijjwqdv0c0yrfnazabwq8p09gcz1qxklcp";
|
||||||
};
|
};
|
||||||
|
|
||||||
buildInputs = [ nose pillow gfortran glibcLocales ];
|
buildInputs = [ pillow gfortran glibcLocales ];
|
||||||
propagatedBuildInputs = [ numpy scipy numpy.blas ];
|
propagatedBuildInputs = [ numpy scipy numpy.blas ];
|
||||||
|
checkInputs = [ pytest ];
|
||||||
|
|
||||||
LC_ALL="en_US.UTF-8";
|
LC_ALL="en_US.UTF-8";
|
||||||
|
|
||||||
# Disable doctests on OSX: https://github.com/scikit-learn/scikit-learn/issues/10213
|
# Skip test_feature_importance_regression - does web fetch
|
||||||
# Disable doctests everywhere: https://github.com/NixOS/nixpkgs/issues/35436
|
|
||||||
checkPhase = ''
|
checkPhase = ''
|
||||||
HOME=$TMPDIR OMP_NUM_THREADS=1 nosetests --doctest-options=+SKIP $out/${python.sitePackages}/sklearn/
|
cd $TMPDIR
|
||||||
|
HOME=$TMPDIR OMP_NUM_THREADS=1 pytest -k "not test_feature_importance_regression" --pyargs sklearn
|
||||||
'';
|
'';
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
meta = with stdenv.lib; {
|
meta = with stdenv.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";
|
||||||
homepage = http://scikit-learn.org;
|
homepage = http://scikit-learn.org;
|
||||||
|
|
Loading…
Reference in New Issue