Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.1 kB · 58 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859{ lib, buildPythonPackage, pythonOlder, fetchFromGitHub, setuptools, absl-py, cloudpickle, dm-tree, jax, jaxlib, numpy, pytestCheckHook, toolz, typing-extensions,}:
buildPythonPackage rec { pname = "chex"; version = "0.1.86"; pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub { owner = "deepmind"; repo = "chex"; rev = "refs/tags/v${version}"; hash = "sha256-Z5Ns4fG5pC99I4xdGjDMKX6YZpTtd1y0TWcIOtr7dug="; };
build-system = [ setuptools ];
dependencies = [ 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 ]; };}