Merge pull request #255992 from samuela/samuela/jax

python3Packages.jax: remove pytest-xdist
This commit is contained in:
Samuel Ainsworth 2023-09-21 17:40:08 -04:00 committed by GitHub
commit 42efd8d712
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -12,7 +12,6 @@
, numpy , numpy
, opt-einsum , opt-einsum
, pytestCheckHook , pytestCheckHook
, pytest-xdist
, pythonOlder , pythonOlder
, scipy , scipy
, stdenv , stdenv
@ -62,18 +61,13 @@ buildPythonPackage rec {
jaxlib' jaxlib'
matplotlib matplotlib
pytestCheckHook pytestCheckHook
pytest-xdist
]; ];
# high parallelism will result in the tests getting stuck
dontUsePytestXdist = true;
# NOTE: Don't run the tests in the expiremental directory as they require flax # NOTE: Don't run the tests in the expiremental directory as they require flax
# which creates a circular dependency. See https://discourse.nixos.org/t/how-to-nix-ify-python-packages-with-circular-dependencies/14648/2. # which creates a circular dependency. See https://discourse.nixos.org/t/how-to-nix-ify-python-packages-with-circular-dependencies/14648/2.
# Not a big deal, this is how the JAX docs suggest running the test suite # Not a big deal, this is how the JAX docs suggest running the test suite
# anyhow. # anyhow.
pytestFlagsArray = [ pytestFlagsArray = [
"--numprocesses=4"
"-W ignore::DeprecationWarning" "-W ignore::DeprecationWarning"
"tests/" "tests/"
]; ];