python3Packages.elegy: jaxlib belongs in buildInputs

2022-01-27 20:51:58 +00:00 · 2022-01-27 20:51:58 +00:00 · 9d9a54adea
parent 97a90b8d6f
commit 9d9a54adea
1 changed files with 3 additions and 0 deletions
--- a/pkgs/development/python-modules/elegy/default.nix
+++ b/pkgs/development/python-modules/elegy/default.nix
@ -4,6 +4,7 @@
 , deepmerge
 , dm-haiku
 , fetchFromGitHub
+, jaxlib
 , lib
 , poetry
 , pytestCheckHook
@ -35,6 +36,8 @@ buildPythonPackage rec {
    poetry
  ];

+  buildInputs = [ jaxlib ];
+
  propagatedBuildInputs = [
    cloudpickle
    deepdish