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