Laksh recursionaut commited on
Commit
e635625
·
unverified ·
1 Parent(s): 6877289

adding deps reconstruction notebook (#17)

Browse files

Co-authored-by: Laksh <laksh.arumugam@recursionpharma.com>

Files changed (2) hide show
  1. requirements.in +3 -0
  2. 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 pytest
 
 
 
 
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
- frozenlist==1.4.1
 
 
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.1
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
- networkx==3.4.2
79
- # via torch
80
- numcodecs==0.13.1
 
 
 
 
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 torchvision
 
 
 
 
 
 
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.0
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.1
167
  # via transformers
168
  tomli==2.0.2
169
  # via pytest
170
- torch==2.5.0
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.0
179
  # via
180
  # -r requirements.in
181
  # pytorch-lightning
182
- torchvision==0.20.0
183
  # via
184
  # -r requirements.in
185
  # timm
186
- tqdm==4.66.5
 
 
 
 
187
  # via
188
  # -r requirements.in
189
  # huggingface-hub
190
  # pytorch-lightning
191
  # transformers
192
- transformers==4.45.2
 
 
 
 
 
 
 
 
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
- xformers==0.0.28.post2
 
 
206
  # via -r requirements.in
207
- yarl==1.16.0
208
  # via aiohttp
209
- zarr==2.18.3
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: