diff options
Diffstat (limited to 'pkgs/development/python-modules/torchgpipe/default.nix')
-rw-r--r-- | pkgs/development/python-modules/torchgpipe/default.nix | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/pkgs/development/python-modules/torchgpipe/default.nix b/pkgs/development/python-modules/torchgpipe/default.nix index 68e113ce70fe..8bde4937eaec 100644 --- a/pkgs/development/python-modules/torchgpipe/default.nix +++ b/pkgs/development/python-modules/torchgpipe/default.nix @@ -1,8 +1,7 @@ { lib , buildPythonPackage , fetchFromGitHub -, isPy27 -, pytest-runner +, pythonOlder , pytestCheckHook , torch }: @@ -11,7 +10,7 @@ buildPythonPackage rec { pname = "torchgpipe"; version = "0.0.7"; - disabled = isPy27; + disabled = pythonOlder "3.6"; src = fetchFromGitHub { owner = "kakaobrain"; @@ -20,9 +19,14 @@ buildPythonPackage rec { sha256 = "0ki0njhmz1i3pkpr3y6h6ac7p5qh1kih06mknc2s18mfw34f2l55"; }; + postPatch = '' + substituteInPlace setup.py \ + --replace "'pytest-runner'" "" + ''; + propagatedBuildInputs = [ torch ]; - checkInputs = [ pytest-runner pytestCheckHook ]; + checkInputs = [ pytestCheckHook ]; disabledTests = [ "test_inplace_on_requires_grad" "test_input_requiring_grad" |