python3Packages.jax: test with limited parallelism

The tests are prone to getting stuck with high parallelism.
2022-04-15 01:39:53 +02:00
parent 214d151733
commit d57404ea3d
1 changed files with 4 additions and 1 deletions
@@ -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/"
  ];