diff options
author | Samuel Ainsworth <skainsworth@gmail.com> | 2022-01-27 20:50:30 +0000 |
---|---|---|
committer | Samuel Ainsworth <skainsworth@gmail.com> | 2022-01-27 20:50:30 +0000 |
commit | 6c2acb04c244f6065f223072cd6b1829114560b7 (patch) | |
tree | e5e83e1782744281110e3df1119c9a9eff087048 /pkgs/development/python-modules/dm-haiku | |
parent | 1387fd0fd14d2ba0c948cd4f27e2a9366bffba3c (diff) |
python3Packages.dm-haiku: jaxlib is a test dependency
Diffstat (limited to 'pkgs/development/python-modules/dm-haiku')
-rw-r--r-- | pkgs/development/python-modules/dm-haiku/default.nix | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/dm-haiku/default.nix b/pkgs/development/python-modules/dm-haiku/default.nix index 5468776c72d11..14ceebf0a19b3 100644 --- a/pkgs/development/python-modules/dm-haiku/default.nix +++ b/pkgs/development/python-modules/dm-haiku/default.nix @@ -4,6 +4,7 @@ , dill , dm-tree , fetchFromGitHub +, jaxlib , jmp , lib , pytestCheckHook @@ -31,6 +32,7 @@ buildPythonPackage rec { chex cloudpickle dm-tree + jaxlib pytestCheckHook tensorflow ]; |