Merge pull request #258122 from mweinelt/jax-tests
python310Packages.jax: reenable pytest-xdist
This commit is contained in:
commit
efb9f8b2dc
|
@ -12,6 +12,7 @@
|
|||
, numpy
|
||||
, opt-einsum
|
||||
, pytestCheckHook
|
||||
, pytest-xdist
|
||||
, pythonOlder
|
||||
, scipy
|
||||
, stdenv
|
||||
|
@ -61,13 +62,18 @@ buildPythonPackage rec {
|
|||
jaxlib'
|
||||
matplotlib
|
||||
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
|
||||
# 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 = [
|
||||
"--numprocesses=4"
|
||||
"-W ignore::DeprecationWarning"
|
||||
"tests/"
|
||||
];
|
||||
|
|
Loading…
Reference in a new issue