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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
{ stdenv
, buildPythonPackage
, dm-haiku
, chex
, cloudpickle
, dill
, dm-tree
, jaxlib
, pytest-xdist
, pytestCheckHook
, tensorflow
, bsuite
, frozendict
, dm-env
, scikitimage
, rlax
, distrax
, tensorflow-probability
, optax }:
buildPythonPackage rec {
pname = "dm-haiku-tests";
inherit (dm-haiku) version;
src = dm-haiku.testsout;
dontBuild = true;
dontInstall = true;
checkInputs = [
bsuite
chex
cloudpickle
dill
distrax
dm-env
dm-haiku
dm-tree
frozendict
jaxlib
pytest-xdist
pytestCheckHook
optax
rlax
scikitimage
tensorflow
tensorflow-probability
];
disabledTests = [
# See https://github.com/deepmind/dm-haiku/issues/366.
"test_jit_Recurrent"
# Assertion errors
"test_connect_conv_padding_function_same0"
"test_connect_conv_padding_function_valid0"
"test_connect_conv_padding_function_same1"
"test_connect_conv_padding_function_same2"
"test_connect_conv_padding_function_valid1"
"test_connect_conv_padding_function_valid2"
"test_invalid_axis_ListString"
"test_invalid_axis_String"
"test_simple_case"
"test_simple_case_with_scale"
"test_slice_axis"
"test_zero_inputs"
];
}
|