python3Packages.chainer: refactor pytest

wip/yesman
ishiy 3 years ago
parent 8798f2c09a
commit c355637b4f
  1. 15
      pkgs/development/python-modules/chainer/default.nix

@ -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…
Cancel
Save