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