python312Packages.chex: 0.1.86 -> 0.1.87 (#345526)

{

  lib,

  buildPythonPackage,

  pythonOlder,

  fetchFromGitHub,

  # build-system

  setuptools,

  # dependencies

  absl-py,

  cloudpickle,

  dm-tree,

  jax,

  jaxlib,

  numpy,

  pytestCheckHook,

  toolz,

  typing-extensions,

  # tests

  cloudpickle,

  dm-tree,

  pytestCheckHook,

}:

buildPythonPackage rec {

  pname = "chex";

  version = "0.1.86";

  version = "0.1.87";

  pyproject = true;

  disabled = pythonOlder "3.9";

  src = fetchFromGitHub {

    owner = "deepmind";

    repo = "chex";

    rev = "refs/tags/v${version}";

    hash = "sha256-Z5Ns4fG5pC99I4xdGjDMKX6YZpTtd1y0TWcIOtr7dug=";

    hash = "sha256-TPh7XLWHk0y/VLXxHLANUiDmfveHPeMLks9QKf16doo=";

  };

  build-system = [ setuptools ];

  dependencies = [

    absl-py

    jaxlib

    jax

    jaxlib

    numpy

    toolz

    typing-extensions

    pytestCheckHook

  ];

  meta = with lib; {

  meta = {

    description = "Chex is a library of utilities for helping to write reliable JAX code";

    homepage = "https://github.com/deepmind/chex";

    changelog = "https://github.com/google-deepmind/chex/releases/tag/v${version}";

    license = licenses.asl20;

    maintainers = with maintainers; [ ndl ];

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ ndl ];

  };

}