python3Packages.jax: fix build

Fix https://github.com/NixOS/nixpkgs/issues/183173
This commit is contained in:
Samuel Ainsworth 2022-08-04 00:16:08 +00:00
parent e0a5ce3f06
commit f32f8e2a57

View File

@ -87,6 +87,19 @@ buildPythonPackage rec {
"testEigvalsGrad_shape"
];
# See https://github.com/google/jax/issues/11722. This is a temporary fix in
# order to unblock etils, and upgrading jax/jaxlib to the latest version. See
# https://github.com/NixOS/nixpkgs/issues/183173#issuecomment-1204074993.
disabledTestPaths = [
"tests/api_test.py"
"tests/core_test.py"
"tests/lax_numpy_indexing_test.py"
"tests/lax_numpy_test.py"
"tests/nn_test.py"
"tests/random_test.py"
"tests/sparse_test.py"
];
pythonImportsCheck = [
"jax"
];