diff options
| author | Samuel Ainsworth <skainsworth@gmail.com> | 2023-09-18 17:48:17 -0700 |
|---|---|---|
| committer | Samuel Ainsworth <skainsworth@gmail.com> | 2023-09-18 17:48:17 -0700 |
| commit | bddafba6a54a4ea07428a6ff0713898f81b25cca (patch) | |
| tree | 47c9bb6fb508c103795acf8053b3e9e2449cfa9c | |
| parent | Merge pull request #255931 from fabaff/google-cloud-dataproc-bump (diff) | |
| download | nixpkgs-origin/samuela/jax.tar.gz | |
python3Packages.jax: remove pytest-xdistorigin/samuela/jax
This dependency has caused nothing but trouble for us since its introduction. It offers moderate test suite speedups, but at the cost of frequent OOM failures in CI, including Hydra and nixpkgs-upkeep. Furthermore, the test suite takes only 1.5 CPU-hrs, limiting xdist's utility.
| -rw-r--r-- | pkgs/development/python-modules/jax/default.nix | 6 |
1 files changed, 0 insertions, 6 deletions
diff --git a/pkgs/development/python-modules/jax/default.nix b/pkgs/development/python-modules/jax/default.nix index b22d82d7f22f..caabb250d992 100644 --- a/pkgs/development/python-modules/jax/default.nix +++ b/pkgs/development/python-modules/jax/default.nix @@ -12,7 +12,6 @@ , numpy , opt-einsum , pytestCheckHook -, pytest-xdist , pythonOlder , scipy , stdenv @@ -58,18 +57,13 @@ 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/" ]; |