Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 77 lines
Nix
at litex
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778{ 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 ]; };}