python3Packages.dm-haiku: jaxlib is a test dependency

2022-01-27 20:50:30 +00:00 · 2022-01-27 20:50:30 +00:00 · 6c2acb04c2
commit 6c2acb04c2
parent 1387fd0fd1
1 changed files with 2 additions and 0 deletions
--- 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
  ];