python3Packages.optax: jaxlib belongs in buildInputs

2022-01-27 20:52:20 +00:00
parent c674a64ce6
commit 236bab3f1f
1 changed files with 2 additions and 1 deletions
@@ -22,10 +22,11 @@ buildPythonPackage rec {
    sha256 = "1q8cxc42a5xais2ll1l238cnn3l7w28savhgiz0lg01ilz2ysbli";
  };

+  buildInputs = [ jaxlib ];
+
  propagatedBuildInputs = [
    absl-py
    chex
-    jaxlib
    numpy
  ];