test.txt 24.8 KB
Newer Older
1
# This file was autogenerated by uv via the following command:
2
#    uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu128 --python-platform x86_64-manylinux_2_28
3
4
5
6
7
8
absl-py==2.1.0
    # via rouge-score
accelerate==1.0.1
    # via
    #   lm-eval
    #   peft
9
10
11
12
aenum==3.1.16
    # via lightly
affine==2.4.0
    # via rasterio
13
aiohappyeyeballs==2.6.1
14
    # via aiohttp
15
aiohttp==3.13.0
16
    # via
Rui Qiao's avatar
Rui Qiao committed
17
    #   aiohttp-cors
18
19
    #   datasets
    #   fsspec
20
    #   gpt-oss
21
    #   lm-eval
Rui Qiao's avatar
Rui Qiao committed
22
23
24
    #   ray
aiohttp-cors==0.8.1
    # via ray
25
aiosignal==1.4.0
Rui Qiao's avatar
Rui Qiao committed
26
    # via aiohttp
27
28
29
30
31
32
albucore==0.0.16
    # via terratorch
albumentations==1.4.6
    # via terratorch
alembic==1.16.4
    # via mlflow
33
34
annotated-types==0.7.0
    # via pydantic
35
36
37
38
antlr4-python3-runtime==4.9.3
    # via
    #   hydra-core
    #   omegaconf
39
anyio==4.6.2.post1
40
41
42
    # via
    #   httpx
    #   starlette
43
44
argcomplete==3.5.1
    # via datamodel-code-generator
45
46
arrow==1.3.0
    # via isoduration
47
48
49
attrs==24.2.0
    # via
    #   aiohttp
50
    #   fiona
51
    #   hypothesis
52
53
    #   jsonlines
    #   jsonschema
54
    #   pytest-subtests
55
    #   rasterio
56
57
58
    #   referencing
audioread==3.0.1
    # via librosa
59
backoff==2.2.1
60
61
62
    # via
    #   -r requirements/test.in
    #   schemathesis
63
bitsandbytes==0.46.1
64
65
66
    # via
    #   -r requirements/test.in
    #   lightning
67
68
black==24.10.0
    # via datamodel-code-generator
69
70
blinker==1.9.0
    # via flask
71
72
blobfile==3.0.0
    # via -r requirements/test.in
73
74
bm25s==0.2.13
    # via mteb
75
boto3==1.35.57
76
77
78
    # via
    #   runai-model-streamer-s3
    #   tensorizer
79
botocore==1.35.57
80
81
82
    # via
    #   boto3
    #   s3transfer
83
84
bounded-pool-executor==0.0.3
    # via pqdm
85
buildkite-test-collector==0.1.9
86
    # via -r requirements/test.in
Rui Qiao's avatar
Rui Qiao committed
87
cachetools==5.5.2
88
89
90
    # via
    #   google-auth
    #   mlflow-skinny
91
92
certifi==2024.8.30
    # via
93
    #   fiona
94
95
    #   httpcore
    #   httpx
96
97
98
99
    #   lightly
    #   pyogrio
    #   pyproj
    #   rasterio
100
101
102
103
104
105
106
    #   requests
cffi==1.17.1
    # via soundfile
chardet==5.2.0
    # via mbstrdecoder
charset-normalizer==3.4.0
    # via requests
107
108
chz==0.3.0
    # via gpt-oss
109
110
111
click==8.1.7
    # via
    #   black
112
113
114
115
    #   click-plugins
    #   cligj
    #   fiona
    #   flask
116
    #   jiwer
117
    #   mlflow-skinny
118
    #   nltk
119
    #   rasterio
120
    #   ray
121
    #   schemathesis
122
    #   typer
123
124
125
126
127
128
129
130
131
132
133
    #   uvicorn
click-plugins==1.1.1.2
    # via
    #   fiona
    #   rasterio
cligj==0.7.2
    # via
    #   fiona
    #   rasterio
cloudpickle==3.1.1
    # via mlflow-skinny
134
135
136
colorama==0.4.6
    # via
    #   sacrebleu
137
    #   schemathesis
138
    #   tqdm-multiprocess
Rui Qiao's avatar
Rui Qiao committed
139
140
colorful==0.5.6
    # via ray
141
142
contourpy==1.3.0
    # via matplotlib
143
144
coverage==7.10.6
    # via pytest-cov
145
146
cramjam==2.9.0
    # via fastparquet
147
cupy-cuda12x==13.6.0
148
149
150
    # via ray
cycler==0.12.1
    # via matplotlib
151
152
databricks-sdk==0.59.0
    # via mlflow-skinny
153
datamodel-code-generator==0.26.3
154
    # via -r requirements/test.in
155
156
157
158
159
160
161
162
dataproperty==1.0.1
    # via
    #   pytablewriter
    #   tabledata
datasets==3.0.2
    # via
    #   evaluate
    #   lm-eval
163
    #   mteb
164
165
decorator==5.1.1
    # via librosa
166
167
decord==0.6.0
    # via -r requirements/test.in
168
169
170
171
172
173
dill==0.3.8
    # via
    #   datasets
    #   evaluate
    #   lm-eval
    #   multiprocess
Rui Qiao's avatar
Rui Qiao committed
174
175
distlib==0.3.9
    # via virtualenv
176
177
dnspython==2.7.0
    # via email-validator
178
docker==7.1.0
179
180
181
    # via
    #   gpt-oss
    #   mlflow
182
183
docopt==0.6.2
    # via num2words
184
185
186
187
188
docstring-parser==0.17.0
    # via jsonargparse
efficientnet-pytorch==0.7.1
    # via segmentation-models-pytorch
einops==0.8.1
189
    # via
190
    #   -r requirements/test.in
191
    #   encodec
192
193
    #   terratorch
    #   torchgeo
194
195
196
197
    #   vector-quantize-pytorch
    #   vocos
einx==0.3.0
    # via vector-quantize-pytorch
198
199
email-validator==2.2.0
    # via pydantic
200
201
encodec==0.1.1
    # via vocos
202
203
eval-type-backport==0.2.2
    # via mteb
204
205
evaluate==0.4.3
    # via lm-eval
206
fastapi==0.116.1
207
208
209
    # via
    #   gpt-oss
    #   mlflow-skinny
210
211
fastparquet==2024.11.0
    # via genai-perf
212
213
fastrlock==0.8.2
    # via cupy-cuda12x
214
215
fastsafetensors==0.1.10
    # via -r requirements/test.in
216
217
filelock==3.16.1
    # via
218
    #   blobfile
219
220
221
222
223
    #   datasets
    #   huggingface-hub
    #   ray
    #   torch
    #   transformers
Rui Qiao's avatar
Rui Qiao committed
224
    #   virtualenv
225
226
227
228
fiona==1.10.1
    # via torchgeo
flask==3.1.1
    # via mlflow
229
fonttools==4.55.0
230
    # via matplotlib
231
232
fqdn==1.5.1
    # via jsonschema
233
234
frozendict==2.4.6
    # via einx
235
236
237
238
frozenlist==1.5.0
    # via
    #   aiohttp
    #   aiosignal
239
fsspec==2024.9.0
240
241
242
    # via
    #   datasets
    #   evaluate
243
    #   fastparquet
244
    #   huggingface-hub
245
246
    #   lightning
    #   pytorch-lightning
247
    #   torch
248
249
ftfy==6.3.1
    # via open-clip-torch
250
genai-perf==0.0.8
251
    # via -r requirements/test.in
252
253
genson==1.3.0
    # via datamodel-code-generator
254
255
256
257
258
259
geopandas==1.0.1
    # via terratorch
gitdb==4.0.12
    # via gitpython
gitpython==3.1.44
    # via mlflow-skinny
Rui Qiao's avatar
Rui Qiao committed
260
google-api-core==2.24.2
261
262
263
264
    # via
    #   google-cloud-core
    #   google-cloud-storage
    #   opencensus
Rui Qiao's avatar
Rui Qiao committed
265
google-auth==2.40.2
266
267
268
    # via
    #   databricks-sdk
    #   google-api-core
269
270
271
272
273
274
275
276
277
278
279
280
281
    #   google-cloud-core
    #   google-cloud-storage
    #   runai-model-streamer-gcs
google-cloud-core==2.4.3
    # via google-cloud-storage
google-cloud-storage==3.4.0
    # via runai-model-streamer-gcs
google-crc32c==1.7.1
    # via
    #   google-cloud-storage
    #   google-resumable-media
google-resumable-media==2.7.2
    # via google-cloud-storage
Rui Qiao's avatar
Rui Qiao committed
282
283
googleapis-common-protos==1.70.0
    # via google-api-core
284
285
gpt-oss==0.0.8
    # via -r requirements/test.in
286
287
graphene==3.4.3
    # via mlflow
288
graphql-core==3.2.6
289
290
291
292
293
294
295
296
    # via
    #   graphene
    #   graphql-relay
    #   hypothesis-graphql
graphql-relay==3.2.0
    # via graphene
greenlet==3.2.3
    # via sqlalchemy
Rui Qiao's avatar
Rui Qiao committed
297
298
grpcio==1.71.0
    # via ray
299
300
gunicorn==23.0.0
    # via mlflow
301
h11==0.14.0
302
303
304
305
306
    # via
    #   httpcore
    #   uvicorn
h5py==3.13.0
    # via terratorch
307
308
harfile==0.3.0
    # via schemathesis
309
hf-xet==1.1.7
310
    # via huggingface-hub
311
312
hiredis==3.0.0
    # via tensorizer
313
314
html2text==2025.4.15
    # via gpt-oss
315
316
317
httpcore==1.0.6
    # via httpx
httpx==0.27.2
318
319
320
    # via
    #   -r requirements/test.in
    #   schemathesis
321
huggingface-hub==0.34.3
322
323
324
325
    # via
    #   accelerate
    #   datasets
    #   evaluate
326
    #   open-clip-torch
327
    #   peft
328
    #   segmentation-models-pytorch
329
    #   sentence-transformers
330
    #   terratorch
331
332
333
    #   timm
    #   tokenizers
    #   transformers
334
    #   vocos
335
336
humanize==4.11.0
    # via runai-model-streamer
337
338
339
340
hydra-core==1.3.2
    # via
    #   lightly
    #   lightning
341
342
343
344
345
346
347
348
349
hypothesis==6.131.0
    # via
    #   hypothesis-graphql
    #   hypothesis-jsonschema
    #   schemathesis
hypothesis-graphql==0.11.1
    # via schemathesis
hypothesis-jsonschema==0.23.1
    # via schemathesis
350
351
352
353
354
idna==3.10
    # via
    #   anyio
    #   email-validator
    #   httpx
355
    #   jsonschema
356
357
    #   requests
    #   yarl
358
359
360
361
362
363
364
365
imageio==2.37.0
    # via scikit-image
importlib-metadata==8.7.0
    # via
    #   mlflow-skinny
    #   opentelemetry-api
importlib-resources==6.5.2
    # via typeshed-client
366
367
368
369
inflect==5.6.2
    # via datamodel-code-generator
iniconfig==2.0.0
    # via pytest
370
371
isoduration==20.11.0
    # via jsonschema
372
373
isort==5.13.2
    # via datamodel-code-generator
374
375
itsdangerous==2.2.0
    # via flask
376
jinja2==3.1.6
377
378
    # via
    #   datamodel-code-generator
379
380
    #   flask
    #   mlflow
381
    #   torch
382
jiwer==3.0.5
383
    # via -r requirements/test.in
384
385
386
387
388
389
390
391
392
jmespath==1.0.1
    # via
    #   boto3
    #   botocore
joblib==1.4.2
    # via
    #   librosa
    #   nltk
    #   scikit-learn
393
394
395
396
jsonargparse==4.35.0
    # via
    #   lightning
    #   terratorch
397
398
jsonlines==4.0.0
    # via lm-eval
399
400
jsonpointer==3.0.0
    # via jsonschema
401
jsonschema==4.23.0
402
    # via
403
    #   hypothesis-jsonschema
404
405
    #   mistral-common
    #   ray
406
    #   schemathesis
407
408
jsonschema-specifications==2024.10.1
    # via jsonschema
409
410
junit-xml==1.9
    # via schemathesis
411
412
kaleido==0.2.1
    # via genai-perf
413
414
kiwisolver==1.4.7
    # via matplotlib
415
416
417
418
kornia==0.8.1
    # via torchgeo
kornia-rs==0.1.9
    # via kornia
419
lazy-loader==0.4
420
421
422
    # via
    #   librosa
    #   scikit-image
423
424
425
libnacl==2.1.0
    # via tensorizer
librosa==0.10.2.post1
426
    # via -r requirements/test.in
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
lightly==1.5.20
    # via
    #   terratorch
    #   torchgeo
lightly-utils==0.0.2
    # via lightly
lightning==2.5.1.post0
    # via
    #   terratorch
    #   torchgeo
lightning-utilities==0.14.3
    # via
    #   lightning
    #   pytorch-lightning
    #   torchmetrics
442
llvmlite==0.44.0
443
    # via numba
444
lm-eval @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d
445
    # via -r requirements/test.in
446
lxml==5.3.0
447
448
    # via
    #   blobfile
449
    #   gpt-oss
450
    #   sacrebleu
451
452
453
454
mako==1.3.10
    # via alembic
markdown==3.8.2
    # via mlflow
455
456
markdown-it-py==3.0.0
    # via rich
Huy Do's avatar
Huy Do committed
457
markupsafe==3.0.1
458
    # via
459
    #   flask
460
    #   jinja2
461
    #   mako
462
    #   werkzeug
463
matplotlib==3.9.2
464
465
466
467
468
469
    # via
    #   -r requirements/test.in
    #   lightning
    #   mlflow
    #   pycocotools
    #   torchgeo
470
471
472
473
474
mbstrdecoder==1.1.3
    # via
    #   dataproperty
    #   pytablewriter
    #   typepy
475
476
mdurl==0.1.2
    # via markdown-it-py
Julien Denize's avatar
Julien Denize committed
477
mistral-common==1.8.2
478
    # via -r requirements/test.in
479
480
481
482
mlflow==2.22.0
    # via terratorch
mlflow-skinny==2.22.0
    # via mlflow
483
484
485
486
487
488
489
490
more-itertools==10.5.0
    # via lm-eval
mpmath==1.3.0
    # via sympy
msgpack==1.1.0
    # via
    #   librosa
    #   ray
491
492
mteb==1.38.11
    # via -r requirements/test.in
493
494
495
496
497
498
499
500
multidict==6.1.0
    # via
    #   aiohttp
    #   yarl
multiprocess==0.70.16
    # via
    #   datasets
    #   evaluate
501
502
munch==4.0.0
    # via pretrainedmodels
503
504
505
mypy-extensions==1.0.0
    # via black
networkx==3.2.1
506
507
508
    # via
    #   scikit-image
    #   torch
509
510
nltk==3.9.1
    # via rouge-score
511
512
num2words==0.5.14
    # via -r requirements/test.in
cyyever's avatar
cyyever committed
513
numba==0.61.2
514
515
516
    # via
    #   -r requirements/test.in
    #   librosa
517
518
519
520
numexpr==2.10.1
    # via lm-eval
numpy==1.26.4
    # via
521
    #   -r requirements/test.in
522
    #   accelerate
523
524
    #   albucore
    #   albumentations
525
    #   bitsandbytes
526
    #   bm25s
527
528
529
    #   contourpy
    #   cupy-cuda12x
    #   datasets
530
    #   decord
531
532
    #   einx
    #   encodec
533
    #   evaluate
534
535
    #   fastparquet
    #   genai-perf
536
537
538
    #   geopandas
    #   h5py
    #   imageio
539
    #   librosa
540
541
    #   lightly
    #   lightly-utils
542
    #   matplotlib
543
    #   mistral-common
544
    #   mlflow
545
    #   mteb
546
547
    #   numba
    #   numexpr
548
    #   opencv-python-headless
549
    #   pandas
550
    #   patsy
551
    #   peft
552
553
554
555
    #   pycocotools
    #   pyogrio
    #   rasterio
    #   rioxarray
556
    #   rouge-score
557
    #   runai-model-streamer
558
    #   sacrebleu
559
    #   scikit-image
560
561
    #   scikit-learn
    #   scipy
562
563
    #   segmentation-models-pytorch
    #   shapely
564
    #   soxr
565
    #   statsmodels
566
    #   tensorboardx
567
    #   tensorizer
568
569
570
    #   tifffile
    #   torchgeo
    #   torchmetrics
571
572
    #   torchvision
    #   transformers
573
    #   tritonclient
574
    #   vocos
575
    #   xarray
Huy Do's avatar
Huy Do committed
576
nvidia-cublas-cu12==12.8.4.1
577
578
579
580
    # via
    #   nvidia-cudnn-cu12
    #   nvidia-cusolver-cu12
    #   torch
Huy Do's avatar
Huy Do committed
581
nvidia-cuda-cupti-cu12==12.8.90
582
    # via torch
Huy Do's avatar
Huy Do committed
583
nvidia-cuda-nvrtc-cu12==12.8.93
584
    # via torch
Huy Do's avatar
Huy Do committed
585
nvidia-cuda-runtime-cu12==12.8.90
586
    # via torch
Huy Do's avatar
Huy Do committed
587
nvidia-cudnn-cu12==9.10.2.21
588
    # via torch
Huy Do's avatar
Huy Do committed
589
nvidia-cufft-cu12==11.3.3.83
590
    # via torch
Huy Do's avatar
Huy Do committed
591
nvidia-cufile-cu12==1.13.1.3
592
    # via torch
Huy Do's avatar
Huy Do committed
593
nvidia-curand-cu12==10.3.9.90
594
    # via torch
Huy Do's avatar
Huy Do committed
595
nvidia-cusolver-cu12==11.7.3.90
Huy Do's avatar
Huy Do committed
596
    # via torch
Huy Do's avatar
Huy Do committed
597
nvidia-cusparse-cu12==12.5.8.93
598
599
600
    # via
    #   nvidia-cusolver-cu12
    #   torch
Huy Do's avatar
Huy Do committed
601
nvidia-cusparselt-cu12==0.7.1
Michael Goin's avatar
Michael Goin committed
602
    # via torch
Huy Do's avatar
Huy Do committed
603
nvidia-nccl-cu12==2.27.3
604
    # via torch
Huy Do's avatar
Huy Do committed
605
nvidia-nvjitlink-cu12==12.8.93
606
    # via
Huy Do's avatar
Huy Do committed
607
    #   nvidia-cufft-cu12
608
609
610
    #   nvidia-cusolver-cu12
    #   nvidia-cusparse-cu12
    #   torch
Huy Do's avatar
Huy Do committed
611
nvidia-nvtx-cu12==12.8.90
612
    # via torch
613
614
615
616
omegaconf==2.3.0
    # via
    #   hydra-core
    #   lightning
617
618
open-clip-torch==2.32.0
    # via -r requirements/test.in
619
620
openai-harmony==0.0.4
    # via gpt-oss
Rui Qiao's avatar
Rui Qiao committed
621
622
623
624
opencensus==0.11.4
    # via ray
opencensus-context==0.1.3
    # via opencensus
625
626
627
opencv-python-headless==4.11.0.86
    # via
    #   -r requirements/test.in
628
629
    #   albucore
    #   albumentations
630
    #   mistral-common
631
632
633
opentelemetry-api==1.35.0
    # via
    #   mlflow-skinny
Rui Qiao's avatar
Rui Qiao committed
634
    #   opentelemetry-exporter-prometheus
635
636
    #   opentelemetry-sdk
    #   opentelemetry-semantic-conventions
Rui Qiao's avatar
Rui Qiao committed
637
638
639
640
opentelemetry-exporter-prometheus==0.56b0
    # via ray
opentelemetry-proto==1.36.0
    # via ray
641
opentelemetry-sdk==1.35.0
Rui Qiao's avatar
Rui Qiao committed
642
643
644
645
    # via
    #   mlflow-skinny
    #   opentelemetry-exporter-prometheus
    #   ray
646
647
opentelemetry-semantic-conventions==0.56b0
    # via opentelemetry-sdk
648
packaging==24.2
649
650
651
652
653
654
    # via
    #   accelerate
    #   black
    #   datamodel-code-generator
    #   datasets
    #   evaluate
655
    #   fastparquet
656
657
    #   geopandas
    #   gunicorn
658
    #   huggingface-hub
659
660
    #   hydra-core
    #   kornia
661
    #   lazy-loader
662
663
    #   lightning
    #   lightning-utilities
664
    #   matplotlib
665
    #   mlflow-skinny
666
    #   peft
667
    #   plotly
668
    #   pooch
669
    #   pyogrio
670
671
    #   pytest
    #   pytest-rerunfailures
672
    #   pytorch-lightning
673
    #   ray
674
675
    #   rioxarray
    #   scikit-image
676
    #   statsmodels
677
678
    #   tensorboardx
    #   torchmetrics
679
680
    #   transformers
    #   typepy
681
    #   xarray
682
683
684
685
pandas==2.2.3
    # via
    #   datasets
    #   evaluate
686
687
    #   fastparquet
    #   genai-perf
688
689
    #   geopandas
    #   mlflow
690
    #   statsmodels
691
692
    #   torchgeo
    #   xarray
693
694
695
696
pathspec==0.12.1
    # via black
pathvalidate==3.2.1
    # via pytablewriter
697
698
patsy==1.0.1
    # via statsmodels
699
peft==0.16.0
700
    # via
701
    #   -r requirements/test.in
702
    #   lm-eval
703
pillow==10.4.0
704
    # via
705
    #   genai-perf
706
707
    #   imageio
    #   lightly-utils
708
    #   matplotlib
709
    #   mistral-common
710
711
    #   scikit-image
    #   segmentation-models-pytorch
712
    #   sentence-transformers
713
    #   torchgeo
714
715
716
717
718
    #   torchvision
platformdirs==4.3.6
    # via
    #   black
    #   pooch
Rui Qiao's avatar
Rui Qiao committed
719
    #   virtualenv
720
721
plotly==5.24.1
    # via genai-perf
722
pluggy==1.5.0
723
724
725
    # via
    #   pytest
    #   pytest-cov
726
727
polars==1.29.0
    # via mteb
728
729
730
731
pooch==1.8.2
    # via librosa
portalocker==2.10.1
    # via sacrebleu
732
pqdm==0.2.0
733
    # via -r requirements/test.in
734
735
pretrainedmodels==0.7.4
    # via segmentation-models-pytorch
Rui Qiao's avatar
Rui Qiao committed
736
prometheus-client==0.22.0
Rui Qiao's avatar
Rui Qiao committed
737
738
739
    # via
    #   opentelemetry-exporter-prometheus
    #   ray
740
propcache==0.2.0
741
742
743
    # via
    #   aiohttp
    #   yarl
Rui Qiao's avatar
Rui Qiao committed
744
745
proto-plus==1.26.1
    # via google-api-core
746
747
protobuf==5.28.3
    # via
Rui Qiao's avatar
Rui Qiao committed
748
749
    #   google-api-core
    #   googleapis-common-protos
750
    #   mlflow-skinny
Rui Qiao's avatar
Rui Qiao committed
751
    #   opentelemetry-proto
Rui Qiao's avatar
Rui Qiao committed
752
    #   proto-plus
753
    #   ray
754
    #   tensorboardx
755
756
757
758
759
760
761
762
    #   tensorizer
psutil==6.1.0
    # via
    #   accelerate
    #   peft
    #   tensorizer
py==1.11.0
    # via pytest-forked
Rui Qiao's avatar
Rui Qiao committed
763
764
py-spy==0.4.0
    # via ray
765
pyarrow==18.0.0
766
767
768
    # via
    #   datasets
    #   genai-perf
769
    #   mlflow
Rui Qiao's avatar
Rui Qiao committed
770
771
772
773
774
775
pyasn1==0.6.1
    # via
    #   pyasn1-modules
    #   rsa
pyasn1-modules==0.4.2
    # via google-auth
776
777
pybind11==2.13.6
    # via lm-eval
778
779
pycocotools==2.0.8
    # via terratorch
Patrick von Platen's avatar
Patrick von Platen committed
780
781
pycountry==24.6.1
    # via pydantic-extra-types
782
783
pycparser==2.22
    # via cffi
784
785
pycryptodomex==3.22.0
    # via blobfile
786
pydantic==2.11.7
787
    # via
788
    #   -r requirements/test.in
789
    #   albumentations
790
    #   datamodel-code-generator
791
    #   fastapi
792
    #   gpt-oss
793
    #   lightly
794
    #   mistral-common
795
    #   mlflow-skinny
796
    #   mteb
797
    #   openai-harmony
Patrick von Platen's avatar
Patrick von Platen committed
798
    #   pydantic-extra-types
Rui Qiao's avatar
Rui Qiao committed
799
    #   ray
800
pydantic-core==2.33.2
801
    # via pydantic
Patrick von Platen's avatar
Patrick von Platen committed
802
803
pydantic-extra-types==2.10.5
    # via mistral-common
804
805
pygments==2.18.0
    # via rich
806
807
pyogrio==0.11.0
    # via geopandas
808
pyparsing==3.2.0
809
810
811
812
813
814
815
816
    # via
    #   matplotlib
    #   rasterio
pyproj==3.7.1
    # via
    #   geopandas
    #   rioxarray
    #   torchgeo
817
818
pyrate-limiter==3.7.0
    # via schemathesis
819
820
pystemmer==3.0.0
    # via mteb
821
822
pytablewriter==1.2.0
    # via lm-eval
823
pytest==8.3.5
824
    # via
825
    #   -r requirements/test.in
826
    #   buildkite-test-collector
827
    #   genai-perf
828
    #   pytest-asyncio
829
    #   pytest-cov
830
    #   pytest-forked
831
    #   pytest-mock
832
833
    #   pytest-rerunfailures
    #   pytest-shard
834
    #   pytest-subtests
835
    #   pytest-timeout
836
    #   schemathesis
837
    #   terratorch
838
pytest-asyncio==0.24.0
839
    # via -r requirements/test.in
840
841
pytest-cov==6.3.0
    # via -r requirements/test.in
842
pytest-forked==1.6.0
843
    # via -r requirements/test.in
844
845
pytest-mock==3.14.0
    # via genai-perf
846
pytest-rerunfailures==14.0
847
    # via -r requirements/test.in
848
pytest-shard==0.1.2
849
    # via -r requirements/test.in
850
851
pytest-subtests==0.14.1
    # via schemathesis
852
853
pytest-timeout==2.3.1
    # via -r requirements/test.in
854
855
python-box==7.3.2
    # via terratorch
856
857
python-dateutil==2.9.0.post0
    # via
858
    #   arrow
859
    #   botocore
860
861
    #   graphene
    #   lightly
862
863
864
    #   matplotlib
    #   pandas
    #   typepy
865
866
python-rapidjson==1.20
    # via tritonclient
867
868
869
870
pytorch-lightning==2.5.2
    # via
    #   lightly
    #   lightning
871
872
pytrec-eval-terrier==0.5.7
    # via mteb
873
874
875
876
877
878
879
pytz==2024.2
    # via
    #   pandas
    #   typepy
pyyaml==6.0.2
    # via
    #   accelerate
880
    #   albumentations
881
882
    #   datamodel-code-generator
    #   datasets
883
    #   genai-perf
884
    #   huggingface-hub
885
886
887
888
    #   jsonargparse
    #   lightning
    #   mlflow-skinny
    #   omegaconf
889
    #   peft
890
    #   pytorch-lightning
891
    #   ray
892
    #   responses
893
    #   schemathesis
894
895
    #   timm
    #   transformers
896
    #   vocos
897
898
rapidfuzz==3.12.1
    # via jiwer
899
900
901
902
903
rasterio==1.4.3
    # via
    #   rioxarray
    #   terratorch
    #   torchgeo
Rui Qiao's avatar
Rui Qiao committed
904
ray==2.48.0
905
    # via -r requirements/test.in
906
907
908
909
910
911
912
913
914
redis==5.2.0
    # via tensorizer
referencing==0.35.1
    # via
    #   jsonschema
    #   jsonschema-specifications
regex==2024.9.11
    # via
    #   nltk
915
    #   open-clip-torch
916
917
918
919
920
921
    #   sacrebleu
    #   tiktoken
    #   transformers
requests==2.32.3
    # via
    #   buildkite-test-collector
922
    #   databricks-sdk
923
    #   datasets
924
    #   docker
925
    #   evaluate
Rui Qiao's avatar
Rui Qiao committed
926
    #   google-api-core
927
    #   google-cloud-storage
928
    #   gpt-oss
929
    #   huggingface-hub
930
    #   lightly
931
    #   lm-eval
932
    #   mistral-common
933
    #   mlflow-skinny
934
    #   mteb
935
936
    #   pooch
    #   ray
937
    #   responses
938
939
    #   schemathesis
    #   starlette-testclient
940
941
    #   tiktoken
    #   transformers
942
943
responses==0.25.3
    # via genai-perf
944
945
946
947
rfc3339-validator==0.1.4
    # via jsonschema
rfc3987==1.3.8
    # via jsonschema
948
rich==13.9.4
949
950
    # via
    #   genai-perf
951
    #   lightning
952
    #   mteb
953
    #   typer
954
955
rioxarray==0.19.0
    # via terratorch
956
957
rouge-score==0.1.2
    # via lm-eval
958
rpds-py==0.20.1
959
960
961
    # via
    #   jsonschema
    #   referencing
Rui Qiao's avatar
Rui Qiao committed
962
963
rsa==4.9.1
    # via google-auth
964
965
rtree==1.4.0
    # via torchgeo
966
runai-model-streamer==0.14.0
967
    # via -r requirements/test.in
968
969
runai-model-streamer-gcs==0.14.0
    # via runai-model-streamer
970
971
runai-model-streamer-s3==0.14.0
    # via runai-model-streamer
972
s3transfer==0.10.3
973
    # via boto3
974
975
976
977
978
sacrebleu==2.4.3
    # via lm-eval
safetensors==0.4.5
    # via
    #   accelerate
979
    #   open-clip-torch
980
981
982
    #   peft
    #   timm
    #   transformers
983
984
schemathesis==3.39.15
    # via -r requirements/test.in
985
986
scikit-image==0.25.2
    # via albumentations
987
988
scikit-learn==1.5.2
    # via
989
    #   albumentations
990
991
    #   librosa
    #   lm-eval
992
    #   mlflow
993
    #   mteb
994
995
996
    #   sentence-transformers
scipy==1.13.1
    # via
997
    #   albumentations
998
    #   bm25s
999
    #   librosa
1000
    #   mlflow
1001
    #   mteb
1002
    #   scikit-image
1003
1004
    #   scikit-learn
    #   sentence-transformers
1005
    #   statsmodels
1006
    #   vocos
1007
1008
1009
1010
segmentation-models-pytorch==0.4.0
    # via
    #   terratorch
    #   torchgeo
1011
sentence-transformers==3.2.1
1012
1013
1014
    # via
    #   -r requirements/test.in
    #   mteb
1015
1016
sentencepiece==0.2.0
    # via mistral-common
1017
setuptools==77.0.3
1018
    # via
1019
    #   lightning-utilities
1020
1021
    #   pytablewriter
    #   torch
Huy Do's avatar
Huy Do committed
1022
    #   triton
1023
1024
1025
1026
shapely==2.1.1
    # via
    #   geopandas
    #   torchgeo
1027
1028
shellingham==1.5.4
    # via typer
1029
1030
six==1.16.0
    # via
1031
    #   junit-xml
1032
    #   lightly
Rui Qiao's avatar
Rui Qiao committed
1033
    #   opencensus
1034
    #   python-dateutil
1035
    #   rfc3339-validator
1036
    #   rouge-score
1037
    #   segmentation-models-pytorch
Rui Qiao's avatar
Rui Qiao committed
1038
1039
smart-open==7.1.0
    # via ray
1040
1041
smmap==5.0.2
    # via gitdb
1042
1043
1044
1045
sniffio==1.3.1
    # via
    #   anyio
    #   httpx
1046
1047
sortedcontainers==2.4.0
    # via hypothesis
1048
1049
soundfile==0.12.1
    # via
1050
    #   -r requirements/test.in
1051
    #   librosa
Julien Denize's avatar
Julien Denize committed
1052
    #   mistral-common
1053
soxr==0.5.0.post1
Julien Denize's avatar
Julien Denize committed
1054
1055
1056
    # via
    #   librosa
    #   mistral-common
1057
1058
1059
1060
sqlalchemy==2.0.41
    # via
    #   alembic
    #   mlflow
1061
1062
sqlitedict==2.1.0
    # via lm-eval
1063
1064
sqlparse==0.5.3
    # via mlflow-skinny
1065
1066
starlette==0.46.2
    # via
1067
    #   fastapi
1068
1069
1070
1071
    #   schemathesis
    #   starlette-testclient
starlette-testclient==0.4.1
    # via schemathesis
1072
1073
statsmodels==0.14.4
    # via genai-perf
1074
1075
structlog==25.4.0
    # via gpt-oss
Huy Do's avatar
Huy Do committed
1076
sympy==1.13.3
1077
1078
1079
    # via
    #   einx
    #   torch
1080
1081
1082
1083
tabledata==1.3.3
    # via pytablewriter
tabulate==0.9.0
    # via sacrebleu
1084
1085
tblib==3.1.0
    # via -r requirements/test.in
1086
1087
tcolorpy==0.1.6
    # via pytablewriter
1088
tenacity==9.1.2
1089
    # via
1090
    #   gpt-oss
1091
1092
    #   lm-eval
    #   plotly
1093
1094
tensorboardx==2.6.4
    # via lightning
1095
tensorizer==2.10.1
1096
    # via -r requirements/test.in
1097
1098
termcolor==3.1.0
    # via gpt-oss
1099
terratorch @ git+https://github.com/IBM/terratorch.git@07184fcf91a1324f831ff521dd238d97fe350e3e
1100
    # via -r requirements/test.in
1101
1102
threadpoolctl==3.5.0
    # via scikit-learn
1103
1104
1105
1106
tifffile==2025.3.30
    # via
    #   scikit-image
    #   terratorch
1107
tiktoken==0.12.0
1108
    # via
1109
    #   gpt-oss
1110
1111
    #   lm-eval
    #   mistral-common
Nicolò Lucchesi's avatar
Nicolò Lucchesi committed
1112
timm==1.0.17
1113
1114
1115
    # via
    #   -r requirements/test.in
    #   open-clip-torch
1116
1117
1118
    #   segmentation-models-pytorch
    #   terratorch
    #   torchgeo
1119
tokenizers==0.22.0
1120
1121
1122
    # via
    #   -r requirements/test.in
    #   transformers
1123
1124
1125
1126
tomli==2.2.1
    # via schemathesis
tomli-w==1.2.0
    # via schemathesis
Huy Do's avatar
Huy Do committed
1127
torch==2.8.0+cu128
1128
    # via
1129
    #   -r requirements/test.in
1130
1131
    #   accelerate
    #   bitsandbytes
1132
    #   efficientnet-pytorch
1133
    #   encodec
1134
    #   fastsafetensors
1135
1136
1137
    #   kornia
    #   lightly
    #   lightning
1138
    #   lm-eval
1139
    #   mteb
1140
    #   open-clip-torch
1141
    #   peft
1142
1143
    #   pretrainedmodels
    #   pytorch-lightning
1144
    #   runai-model-streamer
1145
    #   segmentation-models-pytorch
1146
1147
    #   sentence-transformers
    #   tensorizer
1148
    #   terratorch
1149
    #   timm
1150
    #   torchaudio
1151
1152
    #   torchgeo
    #   torchmetrics
1153
    #   torchvision
1154
1155
    #   vector-quantize-pytorch
    #   vocos
Huy Do's avatar
Huy Do committed
1156
torchaudio==2.8.0+cu128
1157
    # via
1158
    #   -r requirements/test.in
1159
1160
    #   encodec
    #   vocos
1161
1162
1163
1164
1165
1166
1167
1168
torchgeo==0.7.0
    # via terratorch
torchmetrics==1.7.4
    # via
    #   lightning
    #   pytorch-lightning
    #   terratorch
    #   torchgeo
Huy Do's avatar
Huy Do committed
1169
torchvision==0.23.0+cu128
Michael Goin's avatar
Michael Goin committed
1170
1171
    # via
    #   -r requirements/test.in
1172
    #   lightly
1173
    #   open-clip-torch
1174
1175
1176
    #   pretrainedmodels
    #   segmentation-models-pytorch
    #   terratorch
Michael Goin's avatar
Michael Goin committed
1177
    #   timm
1178
    #   torchgeo
1179
1180
1181
1182
1183
tqdm==4.66.6
    # via
    #   datasets
    #   evaluate
    #   huggingface-hub
1184
1185
    #   lightly
    #   lightning
1186
    #   lm-eval
1187
    #   mteb
1188
    #   nltk
1189
    #   open-clip-torch
1190
    #   peft
1191
    #   pqdm
1192
1193
1194
    #   pretrainedmodels
    #   pytorch-lightning
    #   segmentation-models-pytorch
1195
1196
1197
1198
1199
    #   sentence-transformers
    #   tqdm-multiprocess
    #   transformers
tqdm-multiprocess==0.0.11
    # via lm-eval
1200
transformers==4.56.2
1201
    # via
1202
    #   -r requirements/test.in
1203
    #   genai-perf
1204
1205
1206
1207
1208
    #   lm-eval
    #   peft
    #   sentence-transformers
    #   transformers-stream-generator
transformers-stream-generator==0.0.5
1209
    # via -r requirements/test.in
Huy Do's avatar
Huy Do committed
1210
triton==3.4.0
1211
    # via torch
1212
1213
tritonclient==2.51.0
    # via
1214
    #   -r requirements/test.in
1215
    #   genai-perf
1216
typepy==1.3.2
1217
1218
1219
1220
    # via
    #   dataproperty
    #   pytablewriter
    #   tabledata
1221
1222
typer==0.15.2
    # via fastsafetensors
1223
1224
types-python-dateutil==2.9.0.20241206
    # via arrow
1225
1226
typeshed-client==2.8.2
    # via jsonargparse
1227
1228
typing-extensions==4.12.2
    # via
1229
    #   aiosignal
1230
1231
    #   albumentations
    #   alembic
1232
    #   chz
1233
1234
    #   fastapi
    #   graphene
1235
1236
    #   huggingface-hub
    #   librosa
1237
1238
    #   lightning
    #   lightning-utilities
1239
    #   mistral-common
1240
    #   mlflow-skinny
1241
    #   mteb
1242
1243
1244
    #   opentelemetry-api
    #   opentelemetry-sdk
    #   opentelemetry-semantic-conventions
1245
    #   pqdm
1246
1247
    #   pydantic
    #   pydantic-core
Patrick von Platen's avatar
Patrick von Platen committed
1248
    #   pydantic-extra-types
1249
1250
    #   pytorch-lightning
    #   sqlalchemy
1251
    #   torch
1252
    #   torchgeo
1253
    #   typer
1254
    #   typeshed-client
1255
1256
1257
    #   typing-inspection
typing-inspection==0.4.1
    # via pydantic
1258
1259
tzdata==2024.2
    # via pandas
1260
1261
uri-template==1.3.0
    # via jsonschema
1262
urllib3==2.2.3
1263
    # via
1264
    #   blobfile
1265
    #   botocore
1266
1267
    #   docker
    #   lightly
1268
    #   requests
1269
1270
    #   responses
    #   tritonclient
1271
uvicorn==0.35.0
1272
1273
1274
    # via
    #   gpt-oss
    #   mlflow-skinny
1275
vector-quantize-pytorch==1.21.2
1276
    # via -r requirements/test.in
Rui Qiao's avatar
Rui Qiao committed
1277
1278
virtualenv==20.31.2
    # via ray
1279
vocos==0.1.0
1280
    # via -r requirements/test.in
1281
1282
wcwidth==0.2.13
    # via ftfy
1283
1284
1285
webcolors==24.11.1
    # via jsonschema
werkzeug==3.1.3
1286
1287
1288
    # via
    #   flask
    #   schemathesis
1289
1290
word2number==1.1
    # via lm-eval
Rui Qiao's avatar
Rui Qiao committed
1291
1292
wrapt==1.17.2
    # via smart-open
1293
1294
xarray==2025.7.1
    # via rioxarray
1295
1296
1297
1298
xxhash==3.5.0
    # via
    #   datasets
    #   evaluate
1299
yarl==1.17.1
1300
1301
1302
    # via
    #   aiohttp
    #   schemathesis
1303
1304
zipp==3.23.0
    # via importlib-metadata
1305
1306
zstandard==0.23.0
    # via lm-eval