b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

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