b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests 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 , buildPythonPackage
, fetchFromGitHub , fetchFromGitHub
, isPy27 , isPy27
, pytest
, pytestrunner , pytestrunner
, pytestCheckHook , pytestCheckHook
, pytorch , pytorch
@ -23,12 +22,11 @@ buildPythonPackage rec {
propagatedBuildInputs = [ pytorch ]; propagatedBuildInputs = [ pytorch ];
checkInputs = [ pytest pytestrunner pytestCheckHook ]; checkInputs = [ pytestrunner pytestCheckHook ];
disabledTests = [ "test_inplace_on_requires_grad" ]; disabledTests = [
# seems like a harmless failure: "test_inplace_on_requires_grad"
## AssertionError: "test_input_requiring_grad"
## 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.'
meta = with lib; { meta = with lib; {
description = "GPipe implemented in Pytorch and optimized for CUDA rather than TPU"; description = "GPipe implemented in Pytorch and optimized for CUDA rather than TPU";