python3Packages.chainer: refactor pytest
This commit is contained in:
parent
8798f2c09a
commit
c355637b4f
|
@ -1,5 +1,5 @@
|
||||||
{ lib, buildPythonPackage, fetchFromGitHub, isPy3k
|
{ lib, buildPythonPackage, fetchFromGitHub, isPy3k
|
||||||
, filelock, protobuf, numpy, pytest, mock, typing-extensions
|
, filelock, protobuf, numpy, pytestCheckHook, mock, typing-extensions
|
||||||
, cupy, cudaSupport ? false
|
, cupy, cudaSupport ? false
|
||||||
}:
|
}:
|
||||||
|
|
||||||
|
@ -17,7 +17,7 @@ buildPythonPackage rec {
|
||||||
};
|
};
|
||||||
|
|
||||||
checkInputs = [
|
checkInputs = [
|
||||||
pytest
|
pytestCheckHook
|
||||||
mock
|
mock
|
||||||
];
|
];
|
||||||
|
|
||||||
|
@ -28,10 +28,13 @@ buildPythonPackage rec {
|
||||||
typing-extensions
|
typing-extensions
|
||||||
] ++ lib.optionals cudaSupport [ cupy ];
|
] ++ lib.optionals cudaSupport [ cupy ];
|
||||||
|
|
||||||
# avoid gpu tests
|
pytestFlagsArray = [ "tests/chainer_tests/utils_tests" ];
|
||||||
checkPhase = ''
|
|
||||||
pytest tests/chainer_tests/utils_tests -k 'not gpu and not cupy and not ideep'
|
disabledTests = [
|
||||||
'';
|
"gpu"
|
||||||
|
"cupy"
|
||||||
|
"ideep"
|
||||||
|
];
|
||||||
|
|
||||||
meta = with lib; {
|
meta = with lib; {
|
||||||
description = "A flexible framework of neural networks for deep learning";
|
description = "A flexible framework of neural networks for deep learning";
|
||||||
|
|
Loading…
Reference in a new issue