python3Packages.jmp: jaxlib is a test dependency

This commit is contained in:
Samuel Ainsworth 2022-01-27 20:50:47 +00:00
parent 6c2acb04c2
commit 501df061ed

View File

@ -19,10 +19,9 @@ buildPythonPackage rec {
sha256 = "0hh4cmp93wjyidj48gh07vhx2kjvpwd23xvy79bsjn5qaaf6q4cm"; sha256 = "0hh4cmp93wjyidj48gh07vhx2kjvpwd23xvy79bsjn5qaaf6q4cm";
}; };
# Wheel requires only `numpy`, but the import needs both `jax` and `jaxlib`. # Wheel requires only `numpy`, but the import needs `jax`.
propagatedBuildInputs = [ propagatedBuildInputs = [
jax jax
jaxlib
]; ];
pythonImportsCheck = [ pythonImportsCheck = [
@ -30,6 +29,7 @@ buildPythonPackage rec {
]; ];
checkInputs = [ checkInputs = [
jaxlib
pytestCheckHook pytestCheckHook
]; ];