JakeHillion/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

python3Packages.chainer: refactor pytest

Browse Source
This commit is contained in:
ishiy 2021-01-29 01:28:25 +09:00
parent 8798f2c09a
commit c355637b4f
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkgs/development/python-modules/chainer/default.nix
View File

@ -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";