python312Packages.equinox: 0.11.5 -> 0.11.6

{

  lib,

  buildPythonPackage,

  pythonOlder,

  fetchFromGitHub,

  # build-system

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 ];

  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}";