33afbf39f6
checkInputs used to be added to nativeBuildInputs. Now we have nativeCheckInputs to do that instead. Doing this treewide change allows to keep hashes identical to before the introduction of nativeCheckInputs.
78 lines
1.4 KiB
Nix
78 lines
1.4 KiB
Nix
{ lib
|
|
, buildPythonPackage
|
|
, chainer
|
|
, fetchFromGitHub
|
|
, hatchling
|
|
, jupyter
|
|
, keras
|
|
#, mxnet
|
|
, nbconvert
|
|
, nbformat
|
|
, nose
|
|
, numpy
|
|
, parameterized
|
|
, pytestCheckHook
|
|
, pythonOlder
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "einops";
|
|
version = "0.6.0";
|
|
format = "pyproject";
|
|
|
|
disabled = pythonOlder "3.7";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "arogozhnikov";
|
|
repo = pname;
|
|
rev = "refs/tags/v${version}";
|
|
hash = "sha256-/bnp8IhDxp8EB/PoW5Dz+7rOru0/odOrts84aq4qyJw=";
|
|
};
|
|
|
|
nativeBuildInputs = [ hatchling ];
|
|
|
|
nativeCheckInputs = [
|
|
chainer
|
|
jupyter
|
|
keras
|
|
# mxnet (has issues with some CPUs, segfault)
|
|
nbconvert
|
|
nbformat
|
|
nose
|
|
numpy
|
|
parameterized
|
|
pytestCheckHook
|
|
];
|
|
|
|
# No CUDA in sandbox
|
|
EINOPS_SKIP_CUPY = 1;
|
|
|
|
preCheck = ''
|
|
export HOME=$(mktemp -d);
|
|
'';
|
|
|
|
pythonImportsCheck = [
|
|
"einops"
|
|
];
|
|
|
|
disabledTests = [
|
|
# Tests are failing as mxnet is not pulled-in
|
|
# https://github.com/NixOS/nixpkgs/issues/174872
|
|
"test_all_notebooks"
|
|
"test_dl_notebook_with_all_backends"
|
|
"test_backends_installed"
|
|
];
|
|
|
|
disabledTestPaths = [
|
|
"tests/test_layers.py"
|
|
];
|
|
|
|
meta = with lib; {
|
|
description = "Flexible and powerful tensor operations for readable and reliable code";
|
|
homepage = "https://github.com/arogozhnikov/einops";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ yl3dy ];
|
|
};
|
|
}
|
|
|