python3Packages.chainer: 6.4.0 -> 6.5.0
This commit is contained in:
parent
9a47e45857
commit
c064c2adc1
@ -1,18 +1,26 @@
|
|||||||
{ stdenv, lib
|
{ lib, buildPythonPackage, fetchFromGitHub, isPy3k
|
||||||
, buildPythonPackage, fetchPypi, isPy3k
|
, filelock, protobuf, numpy, pytest, mock, typing-extensions
|
||||||
, filelock, protobuf, numpy, pytest, mock
|
|
||||||
, cupy, cudaSupport ? false
|
, cupy, cudaSupport ? false
|
||||||
}:
|
}:
|
||||||
|
|
||||||
buildPythonPackage rec {
|
buildPythonPackage rec {
|
||||||
pname = "chainer";
|
pname = "chainer";
|
||||||
version = "6.4.0";
|
version = "6.5.0";
|
||||||
|
disabled = !isPy3k; # python2.7 abandoned upstream
|
||||||
|
|
||||||
src = fetchPypi {
|
# no tests in Pypi tarball
|
||||||
inherit pname version;
|
src = fetchFromGitHub {
|
||||||
sha256 = "dacbcaa361cebdfbf6f212d138570333611b8f5de553093b1752c578b022a774";
|
owner = "chainer";
|
||||||
|
repo = "chainer";
|
||||||
|
rev = "v${version}";
|
||||||
|
sha256 = "0ha9fbl6sa3fbnsz3y1pg335iiskdbxw838m5j06zgzy156zna1x";
|
||||||
};
|
};
|
||||||
|
|
||||||
|
# remove on 7.0 or 6.6 release
|
||||||
|
postPatch = ''
|
||||||
|
sed -i '/typing/d' setup.py
|
||||||
|
'';
|
||||||
|
|
||||||
checkInputs = [
|
checkInputs = [
|
||||||
pytest
|
pytest
|
||||||
mock
|
mock
|
||||||
@ -22,12 +30,15 @@ buildPythonPackage rec {
|
|||||||
filelock
|
filelock
|
||||||
protobuf
|
protobuf
|
||||||
numpy
|
numpy
|
||||||
|
typing-extensions
|
||||||
] ++ lib.optionals cudaSupport [ cupy ];
|
] ++ lib.optionals cudaSupport [ cupy ];
|
||||||
|
|
||||||
# In python3, test was failed...
|
# avoid gpu tests
|
||||||
doCheck = !isPy3k;
|
checkPhase = ''
|
||||||
|
pytest tests/chainer_tests/utils_tests -k 'not gpu and not cupy'
|
||||||
|
'';
|
||||||
|
|
||||||
meta = with stdenv.lib; {
|
meta = with lib; {
|
||||||
description = "A flexible framework of neural networks for deep learning";
|
description = "A flexible framework of neural networks for deep learning";
|
||||||
homepage = https://chainer.org/;
|
homepage = https://chainer.org/;
|
||||||
license = licenses.mit;
|
license = licenses.mit;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user