python310Packages.torchgpipe: remove useless pytest-runner, correct disabled

This commit is contained in:
Sandro Jäckel 2022-08-30 18:22:56 +02:00 committed by Sandro Jäckel
parent d1be027e2d
commit dbfd754359
No known key found for this signature in database
GPG Key ID: 3AF5A43A3EECC2E5

View File

@ -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"