diff options
Diffstat (limited to 'pkgs/development/python-modules/einops/default.nix')
-rw-r--r-- | pkgs/development/python-modules/einops/default.nix | 17 |
1 files changed, 5 insertions, 12 deletions
diff --git a/pkgs/development/python-modules/einops/default.nix b/pkgs/development/python-modules/einops/default.nix index f4364aac830..9cc5de24e0a 100644 --- a/pkgs/development/python-modules/einops/default.nix +++ b/pkgs/development/python-modules/einops/default.nix @@ -4,20 +4,17 @@ , fetchFromGitHub , hatchling , jupyter -, keras - #, mxnet , nbconvert -, nbformat -, nose , numpy , parameterized +, pillow , pytestCheckHook , pythonOlder }: buildPythonPackage rec { pname = "einops"; - version = "0.6.0"; + version = "0.7.0"; format = "pyproject"; disabled = pythonOlder "3.7"; @@ -26,7 +23,7 @@ buildPythonPackage rec { owner = "arogozhnikov"; repo = pname; rev = "refs/tags/v${version}"; - hash = "sha256-/bnp8IhDxp8EB/PoW5Dz+7rOru0/odOrts84aq4qyJw="; + hash = "sha256-wCs3rMnYCk07kJ3iPItxwCQATflKBYHk6tfBCjiF+bc="; }; nativeBuildInputs = [ hatchling ]; @@ -34,18 +31,14 @@ buildPythonPackage rec { nativeCheckInputs = [ chainer jupyter - keras - # mxnet (has issues with some CPUs, segfault) nbconvert - nbformat - nose numpy parameterized + pillow pytestCheckHook ]; - # No CUDA in sandbox - EINOPS_SKIP_CUPY = 1; + env.EINOPS_TEST_BACKENDS = "numpy,chainer"; preCheck = '' export HOME=$(mktemp -d); |