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

python3Packages.optax: run tests in parallel

Browse source
This commit is contained in:
Samuel Ainsworth 2022-03-29 18:14:04 +00:00 committed by Frederik Rietdijk
parent 6fa30f6855
commit 4570907282
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/optax/default.nix
View file

@ -6,6 +6,7 @@
, jaxlib
, lib
, numpy
, pytest-xdist
, pytestCheckHook
}:
@ -32,8 +33,10 @@ buildPythonPackage rec {
checkInputs = [
dm-haiku
pytest-xdist
pytestCheckHook
];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];
pythonImportsCheck = [
"optax"