nixpkgs/pkgs/development/python-modules/chex/default.nix
Guillaume Girol 33afbf39f6 treewide: switch to nativeCheckInputs
checkInputs used to be added to nativeBuildInputs. Now we have
nativeCheckInputs to do that instead. Doing this treewide change allows
to keep hashes identical to before the introduction of
nativeCheckInputs.
2023-01-21 12:00:00 +00:00

54 lines
929 B
Nix

{ absl-py
, buildPythonPackage
, dm-tree
, fetchFromGitHub
, jax
, jaxlib
, lib
, numpy
, pytestCheckHook
, toolz
}:
buildPythonPackage rec {
pname = "chex";
version = "0.1.5";
format = "setuptools";
src = fetchFromGitHub {
owner = "deepmind";
repo = pname;
rev = "refs/tags/v${version}";
hash = "sha256-FYB0IhokM74HBY2wOJYE6xJrHxHHWhDSAZpWFs6HFu0=";
};
propagatedBuildInputs = [
absl-py
dm-tree
jax
numpy
toolz
];
pythonImportsCheck = [
"chex"
];
nativeCheckInputs = [
jaxlib
pytestCheckHook
];
disabledTests = [
# See https://github.com/deepmind/chex/issues/204.
"test_uninspected_checks"
];
meta = with lib; {
description = "Chex is a library of utilities for helping to write reliable JAX code.";
homepage = "https://github.com/deepmind/chex";
license = licenses.asl20;
maintainers = with maintainers; [ ndl ];
};
}