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

pythonPackages.torchgpipe: disable failing test, cleanup

Browse Source
This commit is contained in:
Sandro Jäckel 2021-02-16 03:48:40 +01:00
parent cf41d097c3
commit 879c3b9e07
No known key found for this signature in database
GPG Key ID: 3AF5A43A3EECC2E5
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pkgs/development/python-modules/torchgpipe/default.nix
View File

@ -2,7 +2,6 @@
, buildPythonPackage
, fetchFromGitHub
, isPy27
, pytest
, pytestrunner
, pytestCheckHook
, pytorch
@ -23,12 +22,11 @@ buildPythonPackage rec {
propagatedBuildInputs = [ pytorch ];
checkInputs = [ pytest pytestrunner pytestCheckHook ];
disabledTests = [ "test_inplace_on_requires_grad" ];
# seems like a harmless failure:
## AssertionError:
## Pattern 'a leaf Variable that requires grad has been used in an in-place operation.'
## does not match 'a leaf Variable that requires grad is being used in an in-place operation.'
checkInputs = [ pytestrunner pytestCheckHook ];
disabledTests = [
"test_inplace_on_requires_grad"
"test_input_requiring_grad"
];
meta = with lib; {
description = "GPipe implemented in Pytorch and optimized for CUDA rather than TPU";