adding deps reconstruction notebook (#17)
Browse filesCo-authored-by: Laksh <laksh.arumugam@recursionpharma.com>
- requirements.in +3 -0
- requirements.txt +130 -17
requirements.in
CHANGED
@@ -9,6 +9,9 @@ xformers
|
|
9 |
zarr
|
10 |
hydra-core
|
11 |
pytorch-lightning>=2.1
|
|
|
|
|
|
|
12 |
isort
|
13 |
ruff
|
14 |
pytest
|
|
|
9 |
zarr
|
10 |
hydra-core
|
11 |
pytorch-lightning>=2.1
|
12 |
+
matplotlib
|
13 |
+
scikit-image
|
14 |
+
ipykernel
|
15 |
isort
|
16 |
ruff
|
17 |
pytest
|
requirements.txt
CHANGED
@@ -18,6 +18,8 @@ antlr4-python3-runtime==4.9.3
|
|
18 |
# omegaconf
|
19 |
asciitree==0.3.3
|
20 |
# via zarr
|
|
|
|
|
21 |
async-timeout==4.0.3
|
22 |
# via aiohttp
|
23 |
attrs==24.2.0
|
@@ -26,8 +28,22 @@ certifi==2024.8.30
|
|
26 |
# via requests
|
27 |
charset-normalizer==3.4.0
|
28 |
# via requests
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
exceptiongroup==1.2.2
|
30 |
-
# via
|
|
|
|
|
|
|
|
|
31 |
fasteners==0.19
|
32 |
# via zarr
|
33 |
filelock==3.16.1
|
@@ -36,7 +52,9 @@ filelock==3.16.1
|
|
36 |
# torch
|
37 |
# transformers
|
38 |
# triton
|
39 |
-
|
|
|
|
|
40 |
# via
|
41 |
# aiohttp
|
42 |
# aiosignal
|
@@ -45,7 +63,7 @@ fsspec[http]==2024.10.0
|
|
45 |
# huggingface-hub
|
46 |
# pytorch-lightning
|
47 |
# torch
|
48 |
-
huggingface-hub==0.26.
|
49 |
# via
|
50 |
# -r requirements.in
|
51 |
# timm
|
@@ -57,31 +75,65 @@ idna==3.10
|
|
57 |
# via
|
58 |
# requests
|
59 |
# yarl
|
|
|
|
|
60 |
iniconfig==2.0.0
|
61 |
# via pytest
|
|
|
|
|
|
|
|
|
62 |
isort==5.13.2
|
63 |
# via -r requirements.in
|
|
|
|
|
64 |
jinja2==3.1.4
|
65 |
# via torch
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
66 |
lightning-utilities==0.11.8
|
67 |
# via
|
68 |
# pytorch-lightning
|
69 |
# torchmetrics
|
70 |
markupsafe==3.0.2
|
71 |
# via jinja2
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
mpmath==1.3.0
|
73 |
# via sympy
|
74 |
multidict==6.1.0
|
75 |
# via
|
76 |
# aiohttp
|
77 |
# yarl
|
78 |
-
|
79 |
-
# via
|
80 |
-
|
|
|
|
|
|
|
|
|
81 |
# via zarr
|
82 |
numpy==1.26.4
|
83 |
# via
|
|
|
|
|
|
|
84 |
# numcodecs
|
|
|
|
|
|
|
85 |
# torchmetrics
|
86 |
# torchvision
|
87 |
# transformers
|
@@ -125,19 +177,49 @@ packaging==24.1
|
|
125 |
# via
|
126 |
# huggingface-hub
|
127 |
# hydra-core
|
|
|
|
|
128 |
# lightning-utilities
|
|
|
129 |
# pytest
|
130 |
# pytorch-lightning
|
|
|
131 |
# torchmetrics
|
132 |
# transformers
|
|
|
|
|
|
|
|
|
133 |
pillow==11.0.0
|
134 |
-
# via
|
|
|
|
|
|
|
|
|
|
|
|
|
135 |
pluggy==1.5.0
|
136 |
# via pytest
|
|
|
|
|
137 |
propcache==0.2.0
|
138 |
# via yarl
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
139 |
pytest==8.3.3
|
140 |
# via -r requirements.in
|
|
|
|
|
|
|
|
|
141 |
pytorch-lightning==2.4.0
|
142 |
# via -r requirements.in
|
143 |
pyyaml==6.0.2
|
@@ -147,27 +229,43 @@ pyyaml==6.0.2
|
|
147 |
# pytorch-lightning
|
148 |
# timm
|
149 |
# transformers
|
|
|
|
|
|
|
|
|
150 |
regex==2024.9.11
|
151 |
# via transformers
|
152 |
requests==2.32.3
|
153 |
# via
|
154 |
# huggingface-hub
|
155 |
# transformers
|
156 |
-
ruff==0.7.
|
157 |
# via -r requirements.in
|
158 |
safetensors==0.4.5
|
159 |
# via
|
160 |
# timm
|
161 |
# transformers
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
162 |
sympy==1.13.1
|
163 |
# via torch
|
|
|
|
|
164 |
timm==1.0.11
|
165 |
# via -r requirements.in
|
166 |
-
tokenizers==0.20.
|
167 |
# via transformers
|
168 |
tomli==2.0.2
|
169 |
# via pytest
|
170 |
-
torch==2.5.
|
171 |
# via
|
172 |
# -r requirements.in
|
173 |
# pytorch-lightning
|
@@ -175,38 +273,53 @@ torch==2.5.0
|
|
175 |
# torchmetrics
|
176 |
# torchvision
|
177 |
# xformers
|
178 |
-
torchmetrics==1.5.
|
179 |
# via
|
180 |
# -r requirements.in
|
181 |
# pytorch-lightning
|
182 |
-
torchvision==0.20.
|
183 |
# via
|
184 |
# -r requirements.in
|
185 |
# timm
|
186 |
-
|
|
|
|
|
|
|
|
|
187 |
# via
|
188 |
# -r requirements.in
|
189 |
# huggingface-hub
|
190 |
# pytorch-lightning
|
191 |
# transformers
|
192 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
193 |
# via -r requirements.in
|
194 |
triton==3.1.0
|
195 |
# via torch
|
196 |
typing-extensions==4.12.2
|
197 |
# via
|
198 |
# huggingface-hub
|
|
|
199 |
# lightning-utilities
|
200 |
# multidict
|
201 |
# pytorch-lightning
|
202 |
# torch
|
203 |
urllib3==2.2.3
|
204 |
# via requests
|
205 |
-
|
|
|
|
|
206 |
# via -r requirements.in
|
207 |
-
yarl==1.
|
208 |
# via aiohttp
|
209 |
-
zarr==2.18.
|
210 |
# via -r requirements.in
|
211 |
|
212 |
# The following packages are considered to be unsafe in a requirements file:
|
|
|
18 |
# omegaconf
|
19 |
asciitree==0.3.3
|
20 |
# via zarr
|
21 |
+
asttokens==2.4.1
|
22 |
+
# via stack-data
|
23 |
async-timeout==4.0.3
|
24 |
# via aiohttp
|
25 |
attrs==24.2.0
|
|
|
28 |
# via requests
|
29 |
charset-normalizer==3.4.0
|
30 |
# via requests
|
31 |
+
comm==0.2.2
|
32 |
+
# via ipykernel
|
33 |
+
contourpy==1.3.0
|
34 |
+
# via matplotlib
|
35 |
+
cycler==0.12.1
|
36 |
+
# via matplotlib
|
37 |
+
debugpy==1.8.7
|
38 |
+
# via ipykernel
|
39 |
+
decorator==5.1.1
|
40 |
+
# via ipython
|
41 |
exceptiongroup==1.2.2
|
42 |
+
# via
|
43 |
+
# ipython
|
44 |
+
# pytest
|
45 |
+
executing==2.1.0
|
46 |
+
# via stack-data
|
47 |
fasteners==0.19
|
48 |
# via zarr
|
49 |
filelock==3.16.1
|
|
|
52 |
# torch
|
53 |
# transformers
|
54 |
# triton
|
55 |
+
fonttools==4.54.1
|
56 |
+
# via matplotlib
|
57 |
+
frozenlist==1.5.0
|
58 |
# via
|
59 |
# aiohttp
|
60 |
# aiosignal
|
|
|
63 |
# huggingface-hub
|
64 |
# pytorch-lightning
|
65 |
# torch
|
66 |
+
huggingface-hub==0.26.2
|
67 |
# via
|
68 |
# -r requirements.in
|
69 |
# timm
|
|
|
75 |
# via
|
76 |
# requests
|
77 |
# yarl
|
78 |
+
imageio==2.36.0
|
79 |
+
# via scikit-image
|
80 |
iniconfig==2.0.0
|
81 |
# via pytest
|
82 |
+
ipykernel==6.29.5
|
83 |
+
# via -r requirements.in
|
84 |
+
ipython==8.29.0
|
85 |
+
# via ipykernel
|
86 |
isort==5.13.2
|
87 |
# via -r requirements.in
|
88 |
+
jedi==0.19.1
|
89 |
+
# via ipython
|
90 |
jinja2==3.1.4
|
91 |
# via torch
|
92 |
+
jupyter-client==8.6.3
|
93 |
+
# via ipykernel
|
94 |
+
jupyter-core==5.7.2
|
95 |
+
# via
|
96 |
+
# ipykernel
|
97 |
+
# jupyter-client
|
98 |
+
kiwisolver==1.4.7
|
99 |
+
# via matplotlib
|
100 |
+
lazy-loader==0.4
|
101 |
+
# via scikit-image
|
102 |
lightning-utilities==0.11.8
|
103 |
# via
|
104 |
# pytorch-lightning
|
105 |
# torchmetrics
|
106 |
markupsafe==3.0.2
|
107 |
# via jinja2
|
108 |
+
matplotlib==3.9.2
|
109 |
+
# via -r requirements.in
|
110 |
+
matplotlib-inline==0.1.7
|
111 |
+
# via
|
112 |
+
# ipykernel
|
113 |
+
# ipython
|
114 |
mpmath==1.3.0
|
115 |
# via sympy
|
116 |
multidict==6.1.0
|
117 |
# via
|
118 |
# aiohttp
|
119 |
# yarl
|
120 |
+
nest-asyncio==1.6.0
|
121 |
+
# via ipykernel
|
122 |
+
networkx==3.2.1
|
123 |
+
# via
|
124 |
+
# scikit-image
|
125 |
+
# torch
|
126 |
+
numcodecs==0.12.1
|
127 |
# via zarr
|
128 |
numpy==1.26.4
|
129 |
# via
|
130 |
+
# contourpy
|
131 |
+
# imageio
|
132 |
+
# matplotlib
|
133 |
# numcodecs
|
134 |
+
# scikit-image
|
135 |
+
# scipy
|
136 |
+
# tifffile
|
137 |
# torchmetrics
|
138 |
# torchvision
|
139 |
# transformers
|
|
|
177 |
# via
|
178 |
# huggingface-hub
|
179 |
# hydra-core
|
180 |
+
# ipykernel
|
181 |
+
# lazy-loader
|
182 |
# lightning-utilities
|
183 |
+
# matplotlib
|
184 |
# pytest
|
185 |
# pytorch-lightning
|
186 |
+
# scikit-image
|
187 |
# torchmetrics
|
188 |
# transformers
|
189 |
+
parso==0.8.4
|
190 |
+
# via jedi
|
191 |
+
pexpect==4.9.0
|
192 |
+
# via ipython
|
193 |
pillow==11.0.0
|
194 |
+
# via
|
195 |
+
# imageio
|
196 |
+
# matplotlib
|
197 |
+
# scikit-image
|
198 |
+
# torchvision
|
199 |
+
platformdirs==4.3.6
|
200 |
+
# via jupyter-core
|
201 |
pluggy==1.5.0
|
202 |
# via pytest
|
203 |
+
prompt-toolkit==3.0.48
|
204 |
+
# via ipython
|
205 |
propcache==0.2.0
|
206 |
# via yarl
|
207 |
+
psutil==6.1.0
|
208 |
+
# via ipykernel
|
209 |
+
ptyprocess==0.7.0
|
210 |
+
# via pexpect
|
211 |
+
pure-eval==0.2.3
|
212 |
+
# via stack-data
|
213 |
+
pygments==2.18.0
|
214 |
+
# via ipython
|
215 |
+
pyparsing==3.2.0
|
216 |
+
# via matplotlib
|
217 |
pytest==8.3.3
|
218 |
# via -r requirements.in
|
219 |
+
python-dateutil==2.9.0.post0
|
220 |
+
# via
|
221 |
+
# jupyter-client
|
222 |
+
# matplotlib
|
223 |
pytorch-lightning==2.4.0
|
224 |
# via -r requirements.in
|
225 |
pyyaml==6.0.2
|
|
|
229 |
# pytorch-lightning
|
230 |
# timm
|
231 |
# transformers
|
232 |
+
pyzmq==26.2.0
|
233 |
+
# via
|
234 |
+
# ipykernel
|
235 |
+
# jupyter-client
|
236 |
regex==2024.9.11
|
237 |
# via transformers
|
238 |
requests==2.32.3
|
239 |
# via
|
240 |
# huggingface-hub
|
241 |
# transformers
|
242 |
+
ruff==0.7.2
|
243 |
# via -r requirements.in
|
244 |
safetensors==0.4.5
|
245 |
# via
|
246 |
# timm
|
247 |
# transformers
|
248 |
+
scikit-image==0.24.0
|
249 |
+
# via -r requirements.in
|
250 |
+
scipy==1.13.1
|
251 |
+
# via scikit-image
|
252 |
+
six==1.16.0
|
253 |
+
# via
|
254 |
+
# asttokens
|
255 |
+
# python-dateutil
|
256 |
+
stack-data==0.6.3
|
257 |
+
# via ipython
|
258 |
sympy==1.13.1
|
259 |
# via torch
|
260 |
+
tifffile==2024.8.30
|
261 |
+
# via scikit-image
|
262 |
timm==1.0.11
|
263 |
# via -r requirements.in
|
264 |
+
tokenizers==0.20.2
|
265 |
# via transformers
|
266 |
tomli==2.0.2
|
267 |
# via pytest
|
268 |
+
torch==2.5.1
|
269 |
# via
|
270 |
# -r requirements.in
|
271 |
# pytorch-lightning
|
|
|
273 |
# torchmetrics
|
274 |
# torchvision
|
275 |
# xformers
|
276 |
+
torchmetrics==1.5.1
|
277 |
# via
|
278 |
# -r requirements.in
|
279 |
# pytorch-lightning
|
280 |
+
torchvision==0.20.1
|
281 |
# via
|
282 |
# -r requirements.in
|
283 |
# timm
|
284 |
+
tornado==6.4.1
|
285 |
+
# via
|
286 |
+
# ipykernel
|
287 |
+
# jupyter-client
|
288 |
+
tqdm==4.66.6
|
289 |
# via
|
290 |
# -r requirements.in
|
291 |
# huggingface-hub
|
292 |
# pytorch-lightning
|
293 |
# transformers
|
294 |
+
traitlets==5.14.3
|
295 |
+
# via
|
296 |
+
# comm
|
297 |
+
# ipykernel
|
298 |
+
# ipython
|
299 |
+
# jupyter-client
|
300 |
+
# jupyter-core
|
301 |
+
# matplotlib-inline
|
302 |
+
transformers==4.46.1
|
303 |
# via -r requirements.in
|
304 |
triton==3.1.0
|
305 |
# via torch
|
306 |
typing-extensions==4.12.2
|
307 |
# via
|
308 |
# huggingface-hub
|
309 |
+
# ipython
|
310 |
# lightning-utilities
|
311 |
# multidict
|
312 |
# pytorch-lightning
|
313 |
# torch
|
314 |
urllib3==2.2.3
|
315 |
# via requests
|
316 |
+
wcwidth==0.2.13
|
317 |
+
# via prompt-toolkit
|
318 |
+
xformers==0.0.28.post3
|
319 |
# via -r requirements.in
|
320 |
+
yarl==1.17.1
|
321 |
# via aiohttp
|
322 |
+
zarr==2.18.2
|
323 |
# via -r requirements.in
|
324 |
|
325 |
# The following packages are considered to be unsafe in a requirements file:
|