python3Packages.treex: jaxlib belongs in buildInputs

This commit is contained in:
Samuel Ainsworth 2022-01-27 20:52:33 +00:00
parent 236bab3f1f
commit 93f88be7d1

View File

@ -5,6 +5,7 @@
, fetchFromGitHub , fetchFromGitHub
, flax , flax
, hypothesis , hypothesis
, jaxlib
, keras , keras
, lib , lib
, poetry-core , poetry-core
@ -38,6 +39,8 @@ buildPythonPackage rec {
poetry-core poetry-core
]; ];
buildInputs = [ jaxlib ];
propagatedBuildInputs = [ propagatedBuildInputs = [
einops einops
flax flax