python3Packages.flax: jaxlib belongs in buildInputs

2022-01-27 20:52:07 +00:00
parent 9d9a54adea
commit c674a64ce6
1 changed files with 3 additions and 0 deletions
@@ -1,5 +1,6 @@
 { buildPythonPackage
 , fetchFromGitHub
+, jaxlib
 , keras
 , lib
 , matplotlib
@@ -21,6 +22,8 @@ buildPythonPackage rec {
    sha256 = "0zvq0vl88hiwmss49bnm7gdmndr1dfza2bcs1fj88a9r7w9dmlsr";
  };

+  buildInputs = [ jaxlib ];
+
  propagatedBuildInputs = [
    matplotlib
    msgpack