Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.7 kB · 74 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475{ lib, stdenv, buildPythonPackage, fetchFromGitHub,
# build-system hatchling,
# dependencies jax, jaxtyping, typing-extensions, wadler-lindig,
# tests beartype, optax, pytest-xdist, pytestCheckHook,}:
buildPythonPackage rec { pname = "equinox"; version = "0.13.1"; pyproject = true;
src = fetchFromGitHub { owner = "patrick-kidger"; repo = "equinox"; tag = "v${version}"; hash = "sha256-txgL5a+kKT28gAS8HianBgnnR+J25R2wrpRr8HEWCXA="; };
# Relax speed constraints on tests that can fail on busy builders postPatch = '' substituteInPlace tests/test_while_loop.py \ --replace-fail "speed < 0.1" "speed < 0.5" \ --replace-fail "speed < 0.5" "speed < 1" \ --replace-fail "speed < 1" "speed < 20" \ --replace-fail "speed < 2" "speed < 20" '';
build-system = [ hatchling ];
dependencies = [ jax jaxtyping typing-extensions wadler-lindig ];
nativeCheckInputs = [ beartype optax pytest-xdist pytestCheckHook ];
disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [ # SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated! "test_filter" ];
pythonImportsCheck = [ "equinox" ];
meta = { description = "JAX library based around a simple idea: represent parameterised functions (such as neural networks) as PyTrees"; changelog = "https://github.com/patrick-kidger/equinox/releases/tag/v${version}"; homepage = "https://github.com/patrick-kidger/equinox"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ GaetanLepage ]; };}