python3Packages.dm-haiku: run tests in parallel
This commit is contained in:
parent
3c25104d0e
commit
740a83a9bd
|
@ -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"
|
||||||
|
|
Loading…
Reference in a new issue