Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.0 kB · 56 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657{ lib, buildPythonPackage, pythonOlder, fetchFromGitHub, absl-py, cloudpickle, dm-tree, jax, jaxlib, numpy, pytestCheckHook, toolz, typing-extensions}:
buildPythonPackage rec { pname = "chex"; version = "0.1.85"; format = "setuptools";
disabled = pythonOlder "3.9";
src = fetchFromGitHub { owner = "deepmind"; repo = "chex"; rev = "refs/tags/v${version}"; hash = "sha256-7k/+2dNNbPBXtbabuOEVpAI7T1SuM4JDf074dmTg/vs="; };
propagatedBuildInputs = [ absl-py jaxlib jax numpy toolz typing-extensions ];
pythonImportsCheck = [ "chex" ];
nativeCheckInputs = [ cloudpickle dm-tree pytestCheckHook ];
meta = with lib; { 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 ]; };}