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
, buildPythonPackage
3
, fetchFromGitHub
4
, isPy27
5
-
, pytest
6
, pytestrunner
7
, pytestCheckHook
8
, pytorch
···
23
24
propagatedBuildInputs = [ pytorch ];
25
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.'
32
33
meta = with lib; {
34
description = "GPipe implemented in Pytorch and optimized for CUDA rather than TPU";
···
2
, buildPythonPackage
3
, fetchFromGitHub
4
, isPy27
5
, pytestrunner
6
, pytestCheckHook
7
, pytorch
···
22
23
propagatedBuildInputs = [ pytorch ];
24
25
+
checkInputs = [ pytestrunner pytestCheckHook ];
26
+
disabledTests = [
27
+
"test_inplace_on_requires_grad"
28
+
"test_input_requiring_grad"
29
+
];
30
31
meta = with lib; {
32
description = "GPipe implemented in Pytorch and optimized for CUDA rather than TPU";