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