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