python3Packages.dm-haiku: run tests in parallel

This commit is contained in:
Samuel Ainsworth 2022-03-29 18:13:47 +00:00 committed by Frederik Rietdijk
parent 3c25104d0e
commit 740a83a9bd

View file

@ -7,6 +7,7 @@
, jaxlib , jaxlib
, jmp , jmp
, lib , lib
, pytest-xdist
, pytestCheckHook , pytestCheckHook
, tabulate , tabulate
, tensorflow , tensorflow
@ -33,9 +34,11 @@ buildPythonPackage rec {
cloudpickle cloudpickle
dm-tree dm-tree
jaxlib jaxlib
pytest-xdist
pytestCheckHook pytestCheckHook
tensorflow tensorflow
]; ];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];
pythonImportsCheck = [ pythonImportsCheck = [
"haiku" "haiku"