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

Merge pull request #255992 from samuela/samuela/jax 

python3Packages.jax: remove pytest-xdist
parents 21c1f392 bddafba6

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

+0 −6
Original line number Diff line number Diff line
@@ -12,7 +12,6 @@ 
, numpy

, opt-einsum

, pytestCheckHook

, pytest-xdist

, pythonOlder

, scipy

, stdenv
@@ -62,18 +61,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/"

  ];