Skip to content

Run pytest tests in parallel for dm-haiku, optax, tensorflow-datasets, flax #166163

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
FRidh merged 4 commits into from
Mar 30, 2022
Merged

Run pytest tests in parallel for dm-haiku, optax, tensorflow-datasets, flax #166163

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
3ef623a
python3Packages.dm-haiku: run tests in parallel
samuela Mar 29, 2022
e33cbde
python3Packages.flax: run tests in parallel
samuela Mar 29, 2022
74e077e
python3Packages.optax: run tests in parallel
samuela Mar 29, 2022
b9fd2f5
python3Packages.tensorflow-datasets: run tests in parallel
samuela Mar 29, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions pkgs/development/python-modules/dm-haiku/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
, jaxlib
, jmp
, lib
, pytest-xdist
, pytestCheckHook
, tabulate
, tensorflow
Expand All @@ -33,9 +34,11 @@ buildPythonPackage rec {
cloudpickle
dm-tree
jaxlib
pytest-xdist
pytestCheckHook
tensorflow
];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];

pythonImportsCheck = [
"haiku"
Expand Down
3 changes: 3 additions & 0 deletions pkgs/development/python-modules/flax/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
, msgpack
, numpy
, optax
, pytest-xdist
, pytestCheckHook
, tensorflow
}:
Expand Down Expand Up @@ -37,9 +38,11 @@ buildPythonPackage rec {

checkInputs = [
keras
pytest-xdist
pytestCheckHook
tensorflow
];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];

disabledTestPaths = [
# Docs test, needs extra deps + we're not interested in it.
Expand Down
3 changes: 3 additions & 0 deletions pkgs/development/python-modules/optax/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
, jaxlib
, lib
, numpy
, pytest-xdist
, pytestCheckHook
}:

Expand All @@ -32,8 +33,10 @@ buildPythonPackage rec {

checkInputs = [
dm-haiku
pytest-xdist
pytestCheckHook
];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];

pythonImportsCheck = [
"optax"
Expand Down
3 changes: 3 additions & 0 deletions pkgs/development/python-modules/tensorflow-datasets/default.nix
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
, protobuf
, pycocotools
, pydub
, pytest-xdist
, pytestCheckHook
, requests
, scikitimage
Expand Down Expand Up @@ -88,12 +89,14 @@ buildPythonPackage rec {
pillow
pycocotools
pydub
pytest-xdist
pytestCheckHook
scikitimage
scipy
tensorflow
tifffile
];
pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];

disabledTestPaths = [
# Sandbox violations: network access, filesystem write attempts outside of build dir, ...
Expand Down