python3Packages.treeo: jaxlib is a test dependency

2022-01-27 20:51:07 +00:00 · 2022-01-27 20:51:07 +00:00 · 97a90b8d6f
commit 97a90b8d6f
parent 501df061ed
1 changed files with 2 additions and 2 deletions
--- a/pkgs/development/python-modules/treeo/default.nix
+++ b/pkgs/development/python-modules/treeo/default.nix
@ -22,12 +22,12 @@ buildPythonPackage rec {
    poetry-core
  ];

-  # These deps are not needed for the wheel, but required during the import.
+  # jax is not declared in the dependencies, but is necessary.
  propagatedBuildInputs = [
    jax
-    jaxlib
  ];

+  checkInputs = [ jaxlib ];
  pythonImportsCheck = [
    "treeo"
  ];