Unverified Commit efb9f8b2 authored by Samuel Ainsworth's avatar Samuel Ainsworth Committed by GitHub
Browse files

Merge pull request #258122 from mweinelt/jax-tests 

python310Packages.jax: reenable pytest-xdist
parents ab5557f4 bfd5c260

pkgs/development/python-modules/jax/default.nix

+6 −0
Original line number Diff line number Diff line
@@ -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/"

  ];