Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768{ lib, buildPythonPackage, fetchFromGitHub,
# build-system hatchling,
# dependencies equinox, jax, jaxtyping, typing-extensions,
# tests beartype, pytest, python,}:
buildPythonPackage (finalAttrs: { pname = "lineax"; version = "0.1.1"; pyproject = true; __structuredAttrs = true;
src = fetchFromGitHub { owner = "patrick-kidger"; repo = "lineax"; tag = "v${finalAttrs.version}"; hash = "sha256-qclL/IE/+gLeBL4huy07npXR3sDlbrTlFfib3qVKupk="; };
build-system = [ hatchling ];
dependencies = [ equinox jax jaxtyping typing-extensions ];
pythonImportsCheck = [ "lineax" ];
nativeCheckInputs = [ beartype pytest ];
# Intentionally not using pytest directly as it leads to JAX out-of-memory'ing # https://github.com/patrick-kidger/lineax/blob/1909d190c1963d5f2d991508c1b2714f2266048b/tests/README.md checkPhase = '' runHook preCheck
${python.interpreter} -m tests
runHook postCheck '';
meta = { description = "Linear solvers in JAX and Equinox"; homepage = "https://github.com/patrick-kidger/lineax"; changelog = "https://github.com/patrick-kidger/lineax/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ GaetanLepage ]; };})