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