pyrox.dev
1{
2 buildPythonPackage,
3 dm-haiku,
4 pytest-xdist,
5 pytestCheckHook,
6 tensorflow,
7 tensorflow-datasets,
8 flax,
9 optax,
10}:
11
12buildPythonPackage {
13 format = "setuptools";
14 pname = "optax-tests";
15 inherit (optax) version;
16
17 src = optax.testsout;
18
19 dontBuild = true;
20 dontInstall = true;
21
22 nativeCheckInputs = [
23 dm-haiku
24 pytest-xdist
25 pytestCheckHook
26 tensorflow
27 tensorflow-datasets
28 flax
29 ];
30
31 disabledTestPaths = [
32 # See https://github.com/deepmind/optax/issues/323
33 "examples/lookahead_mnist_test.py"
34 ];
35}