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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
{
lib,
buildPythonPackage,
fetchPypi,
pythonOlder,
flit-core,
# tests
chex,
jaxlib,
pytest-subtests,
pytest-xdist,
pytestCheckHook,
yapf,
# optional
jupyter,
mediapy,
numpy,
importlib-resources,
typing-extensions,
zipp,
absl-py,
tqdm,
dm-tree,
jax,
tensorflow,
}:
buildPythonPackage rec {
pname = "etils";
version = "1.9.4";
pyproject = true;
disabled = pythonOlder "3.10";
src = fetchPypi {
inherit pname version;
hash = "sha256-+tlQQU8KHKWMcMcJFbABT5lT3ZvPiqlRoPdf+b7L6yQ=";
};
nativeBuildInputs = [ flit-core ];
passthru.optional-dependencies = rec {
array-types = enp;
eapp = [
absl-py # FIXME package simple-parsing
] ++ epy;
ecolab = [
jupyter
numpy
mediapy
] ++ enp ++ epy;
edc = epy;
enp = [ numpy ] ++ epy;
epath = [
importlib-resources
typing-extensions
zipp
] ++ epy;
epy = [ typing-extensions ];
etqdm = [
absl-py
tqdm
] ++ epy;
etree = array-types ++ epy ++ enp ++ etqdm;
etree-dm = [ dm-tree ] ++ etree;
etree-jax = [ jax ] ++ etree;
etree-tf = [ tensorflow ] ++ etree;
all =
array-types
++ eapp
++ ecolab
++ edc
++ enp
++ epath
++ epy
++ etqdm
++ etree
++ etree-dm
++ etree-jax
++ etree-tf;
};
pythonImportsCheck = [ "etils" ];
nativeCheckInputs = [
chex
jaxlib
pytest-subtests
pytest-xdist
pytestCheckHook
yapf
] ++ passthru.optional-dependencies.all;
disabledTests = [
"test_public_access" # requires network access
];
doCheck = false; # error: infinite recursion encountered
meta = with lib; {
changelog = "https://github.com/google/etils/blob/v${version}/CHANGELOG.md";
description = "Collection of eclectic utils";
homepage = "https://github.com/google/etils";
license = licenses.asl20;
maintainers = with maintainers; [ mcwitt ];
};
}
|
