Gaetan Lepage 2024-09-14 18:50:13 +02:00
parent dfc36088ba
commit 7df8f71afb

View File

@ -1,7 +1,6 @@
{
lib,
buildPythonPackage,
pythonOlder,
fetchFromGitHub,
# build-system
@ -22,16 +21,14 @@
buildPythonPackage rec {
pname = "equinox";
version = "0.11.5";
version = "0.11.6";
pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub {
owner = "patrick-kidger";
repo = "equinox";
rev = "refs/tags/v${version}";
hash = "sha256-r4HKn+WJmC7BrTeDDAQ1++TpQhhtLcK6HL2CoM/MGx8=";
hash = "sha256-Ee715otEKmXbXwZ2umrJN3HlIjabQao8hl6MRGqJnNw=";
};
build-system = [ hatchling ];
@ -52,28 +49,6 @@ buildPythonPackage rec {
pythonImportsCheck = [ "equinox" ];
disabledTests = [
# For simplicity, JAX has removed its internal frames from the traceback of the following exception.
# https://github.com/patrick-kidger/equinox/issues/716
"test_abstract"
"test_complicated"
"test_grad"
"test_jvp"
"test_mlp"
"test_num_traces"
"test_pytree_in"
"test_simple"
"test_vmap"
# AssertionError: assert 'foo:\n pri...pe=float32)\n' == 'foo:\n pri...pe=float32)\n'
# Also reported in patrick-kidger/equinox#716
"test_backward_nan"
# Flaky test
# See https://github.com/patrick-kidger/equinox/issues/781
"test_traceback_runtime_eqx"
];
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}";