Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.4 kB · 70 lines
Nix
at flake-libs
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071{ lib, buildPythonPackage, fetchFromGitHub,
# build-system hatchling,
# dependencies equinox, jax, jaxtyping, lineax, typing-extensions,
# tests beartype, jaxlib, optax, pytestCheckHook, pytest-xdist,}:
buildPythonPackage rec { pname = "optimistix"; version = "0.0.10"; pyproject = true;
src = fetchFromGitHub { owner = "patrick-kidger"; repo = "optimistix"; tag = "v${version}"; hash = "sha256-stVPHzv0XNd0I31N2Cj0QYrMmhImyx0cablqZfKBFrM="; };
build-system = [ hatchling ];
dependencies = [ equinox jax jaxtyping lineax typing-extensions ];
pythonImportsCheck = [ "optimistix" ];
nativeCheckInputs = [ beartype jaxlib optax pytestCheckHook pytest-xdist ];
pytestFlagsArray = [ # Since jax 0.5.3: # DeprecationWarning: shape requires ndarray or scalar arguments, got <class 'jax._src.api.ShapeDtypeStruct'> at position 0. In a future JAX release this will be an error. "-W" "ignore::DeprecationWarning" ];
meta = { description = "Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox"; homepage = "https://github.com/patrick-kidger/optimistix"; changelog = "https://github.com/patrick-kidger/optimistix/releases/tag/v${version}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ GaetanLepage ]; };}