alioth/nixpkgs
3
0
Fork 0
forked from mirrors/nixpkgs
Code Releases Activity

python3Packages.jax: test with limited parallelism

Browse source 
The tests are prone to getting stuck with high parallelism.
This commit is contained in:
Martin Weinelt 2022-04-10 17:31:15 +02:00
parent 214d151733
commit d57404ea3d
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pkgs/development/python-modules/jax/default.nix
View file

@ -54,12 +54,15 @@ buildPythonPackage rec {
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
# 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
# anyhow.
pytestFlagsArray = [
"-n auto"
"--numprocesses=4"
"-W ignore::DeprecationWarning"
"tests/"
];