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