Something went wrong. Try again.
lol
Something went wrong. Try again.
1.2 kB · 60 lines
Nix
at master
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061{ lib, buildPythonPackage, fetchFromGitHub, hatchling, jupyter, nbconvert, numpy, parameterized, pillow, pytestCheckHook, writableTmpDirAsHomeHook, torch,}:
buildPythonPackage rec { pname = "einops"; version = "0.8.1"; pyproject = true;
src = fetchFromGitHub { owner = "arogozhnikov"; repo = "einops"; tag = "v${version}"; hash = "sha256-J9m5LMOleHf2UziUbOtwf+DFpu/wBDcAyHUor4kqrR8="; };
build-system = [ hatchling ];
nativeCheckInputs = [ writableTmpDirAsHomeHook jupyter nbconvert numpy parameterized pillow pytestCheckHook torch ];
env.EINOPS_TEST_BACKENDS = "numpy";
pythonImportsCheck = [ "einops" ];
disabledTestPaths = [ # skip folder with notebook samples that depend on large packages # or accelerator access and have been unreliable "scripts/" ];
__darwinAllowLocalNetworking = true;
meta = { changelog = "https://github.com/arogozhnikov/einops/releases/tag/${src.tag}"; description = "Flexible and powerful tensor operations for readable and reliable code"; homepage = "https://github.com/arogozhnikov/einops"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ yl3dy ]; };}