cafkafk/nixpkgs
cafkafk/nixpkgs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

python3Packages.flax: run tests in parallel

Browse source
This commit is contained in:
Samuel Ainsworth 2022-03-29 18:13:57 +00:00 committed by Frederik Rietdijk
parent 740a83a9bd
commit 6fa30f6855
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pkgs/development/python-modules/flax/default.nix
View file

@ -7,6 +7,7 @@
, msgpack
, numpy
, optax
, pytest-xdist
, pytestCheckHook
, tensorflow
}:
@ -37,9 +38,11 @@ buildPythonPackage rec {
checkInputs = [
keras
pytest-xdist
pytestCheckHook
tensorflow
];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];
disabledTestPaths = [
# Docs test, needs extra deps + we're not interested in it.