python3Packages.tensorly: 0.4.5 -> 0.7.0
This commit is contained in:
parent
e2b2136d63
commit
1f5e0c848e
@ -1,45 +1,36 @@
|
||||
{ lib
|
||||
, buildPythonPackage
|
||||
, fetchFromGitHub
|
||||
, pytest
|
||||
, nose
|
||||
, pytestCheckHook
|
||||
, isPy27
|
||||
, numpy
|
||||
, scipy
|
||||
, sparse
|
||||
, pytorch
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "tensorly";
|
||||
version = "0.4.5";
|
||||
version = "0.7.0";
|
||||
disabled = isPy27;
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = pname;
|
||||
repo = pname;
|
||||
rev = version;
|
||||
sha256 = "1ml91yaxwx4msisxbm92yf22qfrscvk58f3z2r1jhi96pw2k4i7x";
|
||||
sha256 = "VcX3pCczZQUYZaD7xrrkOcj0QPJt28cYTwpZm5D/X3c=";
|
||||
};
|
||||
|
||||
propagatedBuildInputs = [ numpy scipy sparse ]
|
||||
++ lib.optionals (!doCheck) [ nose ]; # upstream added nose to install_requires
|
||||
|
||||
checkInputs = [ pytest nose pytorch ];
|
||||
# also has a cupy backend, but the tests are currently broken
|
||||
# (e.g. attempts to access cupy.qr instead of cupy.linalg.qr)
|
||||
# and this backend also adds a non-optional CUDA dependence,
|
||||
# as well as tensorflow and mxnet backends, but the tests don't
|
||||
# seem to exercise these backend by default
|
||||
|
||||
# uses >= 140GB of ram to test
|
||||
doCheck = false;
|
||||
checkPhase = ''
|
||||
runHook preCheck
|
||||
nosetests -e "test_cupy"
|
||||
runHook postCheck
|
||||
# nose is not actually required for anything
|
||||
# (including testing with the minimal dependencies)
|
||||
postPatch = ''
|
||||
substituteInPlace setup.py --replace ", 'nose'" ""
|
||||
'';
|
||||
|
||||
propagatedBuildInputs = [ numpy scipy sparse ];
|
||||
|
||||
checkInputs = [ pytestCheckHook ];
|
||||
pytestFlagsArray = [ "tensorly" ];
|
||||
|
||||
pythonImportsCheck = [ "tensorly" ];
|
||||
|
||||
meta = with lib; {
|
||||
|
Loading…
Reference in New Issue
Block a user