nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-02-25 21:44:49 +00:00
Code Issues 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"