python3Packages.treeo: jaxlib is a test dependency

This commit is contained in:
Samuel Ainsworth 2022-01-27 20:51:07 +00:00
parent 501df061ed
commit 97a90b8d6f

View File

@ -22,12 +22,12 @@ buildPythonPackage rec {
poetry-core 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 = [ propagatedBuildInputs = [
jax jax
jaxlib
]; ];
checkInputs = [ jaxlib ];
pythonImportsCheck = [ pythonImportsCheck = [
"treeo" "treeo"
]; ];