pyrox.dev
pythonPackages.torchgpipe: disable failing test, cleanup
+5
-7
pkgs/development/python-modules/torchgpipe/default.nix
+5
-7
pkgs/development/python-modules/torchgpipe/default.nix
···
2
2
, buildPythonPackage
3
3
, fetchFromGitHub
4
4
, isPy27
5
-
, pytest
6
5
, pytestrunner
7
6
, pytestCheckHook
8
7
, pytorch
···
23
22
24
23
propagatedBuildInputs = [ pytorch ];
25
24
26
-
checkInputs = [ pytest pytestrunner pytestCheckHook ];
27
-
disabledTests = [ "test_inplace_on_requires_grad" ];
28
-
# seems like a harmless failure:
29
-
## AssertionError:
30
-
## Pattern 'a leaf Variable that requires grad has been used in an in-place operation.'
31
-
## does not match 'a leaf Variable that requires grad is being used in an in-place operation.'
25
+
checkInputs = [ pytestrunner pytestCheckHook ];
26
+
disabledTests = [
27
+
"test_inplace_on_requires_grad"
28
+
"test_input_requiring_grad"
29
+
];
32
30
33
31
meta = with lib; {
34
32
description = "GPipe implemented in Pytorch and optimized for CUDA rather than TPU";