diff --git a/pkgs/development/python-modules/equinox/default.nix b/pkgs/development/python-modules/equinox/default.nix index 592d10d8f8cd..831241312a9b 100644 --- a/pkgs/development/python-modules/equinox/default.nix +++ b/pkgs/development/python-modules/equinox/default.nix @@ -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}";