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