python3Packages.jmp: jaxlib is a test dependency
This commit is contained in:
parent
6c2acb04c2
commit
501df061ed
@ -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
|
||||||
];
|
];
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user