test.txt 15 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
3
4
5
6
7
8
9
10
absl-py==2.1.0
    # via rouge-score
accelerate==1.0.1
    # via
    #   lm-eval
    #   peft
aiohappyeyeballs==2.4.3
    # via aiohttp
11
aiohttp==3.10.11
12
13
14
15
16
17
18
19
20
21
22
    # via
    #   datasets
    #   fsspec
    #   lm-eval
aiosignal==1.3.1
    # via
    #   aiohttp
    #   ray
annotated-types==0.7.0
    # via pydantic
anyio==4.6.2.post1
23
24
25
    # via
    #   httpx
    #   starlette
26
27
argcomplete==3.5.1
    # via datamodel-code-generator
28
29
arrow==1.3.0
    # via isoduration
30
31
32
attrs==24.2.0
    # via
    #   aiohttp
33
    #   hypothesis
34
35
    #   jsonlines
    #   jsonschema
36
    #   pytest-subtests
37
38
39
    #   referencing
audioread==3.0.1
    # via librosa
40
awscli==1.35.23
41
    # via -r requirements/test.in
42
backoff==2.2.1
43
44
45
    # via
    #   -r requirements/test.in
    #   schemathesis
Michael Goin's avatar
Michael Goin committed
46
bitsandbytes==0.45.3
47
    # via -r requirements/test.in
48
49
black==24.10.0
    # via datamodel-code-generator
50
51
blobfile==3.0.0
    # via -r requirements/test.in
52
boto3==1.35.57
53
    # via tensorizer
54
botocore==1.35.57
55
56
57
58
    # via
    #   awscli
    #   boto3
    #   s3transfer
59
60
bounded-pool-executor==0.0.3
    # via pqdm
61
buildkite-test-collector==0.1.9
62
    # via -r requirements/test.in
63
64
65
66
67
68
69
70
71
72
73
74
75
76
certifi==2024.8.30
    # via
    #   httpcore
    #   httpx
    #   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
77
    #   jiwer
78
79
    #   nltk
    #   ray
80
    #   schemathesis
81
    #   typer
82
83
84
85
colorama==0.4.6
    # via
    #   awscli
    #   sacrebleu
86
    #   schemathesis
87
88
89
    #   tqdm-multiprocess
contourpy==1.3.0
    # via matplotlib
90
91
cramjam==2.9.0
    # via fastparquet
92
93
94
95
cupy-cuda12x==13.3.0
    # via ray
cycler==0.12.1
    # via matplotlib
96
datamodel-code-generator==0.26.3
97
    # via -r requirements/test.in
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
dataproperty==1.0.1
    # via
    #   pytablewriter
    #   tabledata
datasets==3.0.2
    # via
    #   evaluate
    #   lm-eval
decorator==5.1.1
    # via librosa
dill==0.3.8
    # via
    #   datasets
    #   evaluate
    #   lm-eval
    #   multiprocess
dnspython==2.7.0
    # via email-validator
116
117
docopt==0.6.2
    # via num2words
118
119
120
docutils==0.16
    # via awscli
einops==0.8.0
121
    # via
122
    #   -r requirements/test.in
123
    #   encodec
Shinichi Hemmi's avatar
Shinichi Hemmi committed
124
    #   mamba-ssm
125
126
127
128
    #   vector-quantize-pytorch
    #   vocos
einx==0.3.0
    # via vector-quantize-pytorch
129
130
email-validator==2.2.0
    # via pydantic
131
132
encodec==0.1.1
    # via vocos
133
134
evaluate==0.4.3
    # via lm-eval
135
136
fastparquet==2024.11.0
    # via genai-perf
137
138
fastrlock==0.8.2
    # via cupy-cuda12x
139
140
fastsafetensors==0.1.10
    # via -r requirements/test.in
141
142
filelock==3.16.1
    # via
143
    #   blobfile
144
145
146
147
148
149
150
    #   datasets
    #   huggingface-hub
    #   ray
    #   torch
    #   transformers
fonttools==4.54.1
    # via matplotlib
151
152
fqdn==1.5.1
    # via jsonschema
153
154
frozendict==2.4.6
    # via einx
155
156
157
158
159
frozenlist==1.5.0
    # via
    #   aiohttp
    #   aiosignal
    #   ray
160
fsspec==2024.9.0
161
162
163
    # via
    #   datasets
    #   evaluate
164
    #   fastparquet
165
166
    #   huggingface-hub
    #   torch
167
genai-perf==0.0.8
168
    # via -r requirements/test.in
169
170
genson==1.3.0
    # via datamodel-code-generator
171
172
graphql-core==3.2.6
    # via hypothesis-graphql
173
174
h11==0.14.0
    # via httpcore
175
176
harfile==0.3.0
    # via schemathesis
177
178
hf-xet==0.1.4
    # via huggingface-hub
179
180
181
182
183
hiredis==3.0.0
    # via tensorizer
httpcore==1.0.6
    # via httpx
httpx==0.27.2
184
185
186
    # via
    #   -r requirements/test.in
    #   schemathesis
187
huggingface-hub==0.30.1
188
    # via
189
    #   -r requirements/test.in
190
191
192
193
194
195
196
197
    #   accelerate
    #   datasets
    #   evaluate
    #   peft
    #   sentence-transformers
    #   timm
    #   tokenizers
    #   transformers
198
    #   vocos
199
200
humanize==4.11.0
    # via runai-model-streamer
201
202
203
204
205
206
207
208
209
hypothesis==6.131.0
    # via
    #   hypothesis-graphql
    #   hypothesis-jsonschema
    #   schemathesis
hypothesis-graphql==0.11.1
    # via schemathesis
hypothesis-jsonschema==0.23.1
    # via schemathesis
210
211
212
213
214
idna==3.10
    # via
    #   anyio
    #   email-validator
    #   httpx
215
    #   jsonschema
216
217
218
219
220
221
    #   requests
    #   yarl
inflect==5.6.2
    # via datamodel-code-generator
iniconfig==2.0.0
    # via pytest
222
223
isoduration==20.11.0
    # via jsonschema
224
225
isort==5.13.2
    # via datamodel-code-generator
226
jinja2==3.1.6
227
228
229
    # via
    #   datamodel-code-generator
    #   torch
230
jiwer==3.0.5
231
    # via -r requirements/test.in
232
233
234
235
236
237
238
239
240
241
242
jmespath==1.0.1
    # via
    #   boto3
    #   botocore
joblib==1.4.2
    # via
    #   librosa
    #   nltk
    #   scikit-learn
jsonlines==4.0.0
    # via lm-eval
243
244
jsonpointer==3.0.0
    # via jsonschema
245
jsonschema==4.23.0
246
    # via
247
    #   hypothesis-jsonschema
248
249
    #   mistral-common
    #   ray
250
    #   schemathesis
251
252
jsonschema-specifications==2024.10.1
    # via jsonschema
253
254
junit-xml==1.9
    # via schemathesis
255
256
kaleido==0.2.1
    # via genai-perf
257
258
259
260
261
262
263
kiwisolver==1.4.7
    # via matplotlib
lazy-loader==0.4
    # via librosa
libnacl==2.1.0
    # via tensorizer
librosa==0.10.2.post1
264
    # via -r requirements/test.in
265
llvmlite==0.44.0
266
    # via numba
267
lm-eval==0.4.8
268
    # via -r requirements/test.in
269
lxml==5.3.0
270
271
272
    # via
    #   blobfile
    #   sacrebleu
Shinichi Hemmi's avatar
Shinichi Hemmi committed
273
274
mamba-ssm==2.2.4
    # via -r requirements/test.in
275
276
markdown-it-py==3.0.0
    # via rich
277
markupsafe==3.0.2
278
279
280
    # via
    #   jinja2
    #   werkzeug
281
matplotlib==3.9.2
282
    # via -r requirements/test.in
283
284
285
286
287
mbstrdecoder==1.1.3
    # via
    #   dataproperty
    #   pytablewriter
    #   typepy
288
289
mdurl==0.1.2
    # via markdown-it-py
290
mistral-common==1.5.4
291
    # via -r requirements/test.in
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
more-itertools==10.5.0
    # via lm-eval
mpmath==1.3.0
    # via sympy
msgpack==1.1.0
    # via
    #   librosa
    #   ray
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
    # via torch
Shinichi Hemmi's avatar
Shinichi Hemmi committed
312
313
ninja==1.11.1.3
    # via mamba-ssm
314
315
nltk==3.9.1
    # via rouge-score
316
317
num2words==0.5.14
    # via -r requirements/test.in
cyyever's avatar
cyyever committed
318
numba==0.61.2
319
320
321
    # via
    #   -r requirements/test.in
    #   librosa
322
323
324
325
numexpr==2.10.1
    # via lm-eval
numpy==1.26.4
    # via
326
    #   -r requirements/test.in
327
328
329
330
331
    #   accelerate
    #   bitsandbytes
    #   contourpy
    #   cupy-cuda12x
    #   datasets
332
333
    #   einx
    #   encodec
334
    #   evaluate
335
336
    #   fastparquet
    #   genai-perf
337
338
    #   librosa
    #   matplotlib
339
    #   mistral-common
340
341
    #   numba
    #   numexpr
342
    #   opencv-python-headless
343
    #   pandas
344
    #   patsy
345
346
    #   peft
    #   rouge-score
347
    #   runai-model-streamer
348
349
350
351
    #   sacrebleu
    #   scikit-learn
    #   scipy
    #   soxr
352
    #   statsmodels
353
354
355
    #   tensorizer
    #   torchvision
    #   transformers
356
    #   tritonclient
357
    #   vocos
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
nvidia-cublas-cu12==12.4.5.8
    # via
    #   nvidia-cudnn-cu12
    #   nvidia-cusolver-cu12
    #   torch
nvidia-cuda-cupti-cu12==12.4.127
    # via torch
nvidia-cuda-nvrtc-cu12==12.4.127
    # via torch
nvidia-cuda-runtime-cu12==12.4.127
    # via torch
nvidia-cudnn-cu12==9.1.0.70
    # via torch
nvidia-cufft-cu12==11.2.1.3
    # via torch
nvidia-curand-cu12==10.3.5.147
    # via torch
nvidia-cusolver-cu12==11.6.1.9
    # via torch
nvidia-cusparse-cu12==12.3.1.170
    # via
    #   nvidia-cusolver-cu12
    #   torch
Michael Goin's avatar
Michael Goin committed
381
382
nvidia-cusparselt-cu12==0.6.2
    # via torch
383
384
385
386
387
388
389
390
391
nvidia-nccl-cu12==2.21.5
    # via torch
nvidia-nvjitlink-cu12==12.4.127
    # via
    #   nvidia-cusolver-cu12
    #   nvidia-cusparse-cu12
    #   torch
nvidia-nvtx-cu12==12.4.127
    # via torch
392
393
394
395
opencv-python-headless==4.11.0.86
    # via
    #   -r requirements/test.in
    #   mistral-common
396
397
398
399
400
401
402
packaging==24.1
    # via
    #   accelerate
    #   black
    #   datamodel-code-generator
    #   datasets
    #   evaluate
403
    #   fastparquet
404
405
    #   huggingface-hub
    #   lazy-loader
Shinichi Hemmi's avatar
Shinichi Hemmi committed
406
    #   mamba-ssm
407
408
    #   matplotlib
    #   peft
409
    #   plotly
410
411
412
413
    #   pooch
    #   pytest
    #   pytest-rerunfailures
    #   ray
414
    #   statsmodels
415
416
417
418
419
420
    #   transformers
    #   typepy
pandas==2.2.3
    # via
    #   datasets
    #   evaluate
421
422
423
    #   fastparquet
    #   genai-perf
    #   statsmodels
424
425
426
427
pathspec==0.12.1
    # via black
pathvalidate==3.2.1
    # via pytablewriter
428
429
patsy==1.0.1
    # via statsmodels
430
431
peft==0.13.2
    # via
432
    #   -r requirements/test.in
433
    #   lm-eval
434
pillow==10.4.0
435
    # via
436
    #   genai-perf
437
    #   matplotlib
438
    #   mistral-common
439
440
441
442
443
444
    #   sentence-transformers
    #   torchvision
platformdirs==4.3.6
    # via
    #   black
    #   pooch
445
446
plotly==5.24.1
    # via genai-perf
447
448
449
450
451
452
pluggy==1.5.0
    # via pytest
pooch==1.8.2
    # via librosa
portalocker==2.10.1
    # via sacrebleu
453
pqdm==0.2.0
454
    # via -r requirements/test.in
455
456
457
458
459
460
461
462
463
464
465
466
467
468
propcache==0.2.0
    # via yarl
protobuf==5.28.3
    # via
    #   ray
    #   tensorizer
psutil==6.1.0
    # via
    #   accelerate
    #   peft
    #   tensorizer
py==1.11.0
    # via pytest-forked
pyarrow==18.0.0
469
470
471
    # via
    #   datasets
    #   genai-perf
472
473
474
475
476
477
pyasn1==0.6.1
    # via rsa
pybind11==2.13.6
    # via lm-eval
pycparser==2.22
    # via cffi
478
479
pycryptodomex==3.22.0
    # via blobfile
480
pydantic==2.9.2
481
482
483
    # via
    #   datamodel-code-generator
    #   mistral-common
484
485
pydantic-core==2.23.4
    # via pydantic
486
487
pygments==2.18.0
    # via rich
488
489
pyparsing==3.2.0
    # via matplotlib
490
491
pyrate-limiter==3.7.0
    # via schemathesis
492
493
494
495
pytablewriter==1.2.0
    # via lm-eval
pytest==8.3.3
    # via
496
    #   -r requirements/test.in
497
    #   buildkite-test-collector
498
    #   genai-perf
499
500
    #   pytest-asyncio
    #   pytest-forked
501
    #   pytest-mock
502
503
    #   pytest-rerunfailures
    #   pytest-shard
504
    #   pytest-subtests
505
    #   pytest-timeout
506
    #   schemathesis
507
pytest-asyncio==0.24.0
508
    # via -r requirements/test.in
509
pytest-forked==1.6.0
510
    # via -r requirements/test.in
511
512
pytest-mock==3.14.0
    # via genai-perf
513
pytest-rerunfailures==14.0
514
    # via -r requirements/test.in
515
pytest-shard==0.1.2
516
    # via -r requirements/test.in
517
518
pytest-subtests==0.14.1
    # via schemathesis
519
520
pytest-timeout==2.3.1
    # via -r requirements/test.in
521
522
python-dateutil==2.9.0.post0
    # via
523
    #   arrow
524
525
526
527
    #   botocore
    #   matplotlib
    #   pandas
    #   typepy
528
529
python-rapidjson==1.20
    # via tritonclient
530
531
532
533
534
535
536
537
538
539
pytz==2024.2
    # via
    #   pandas
    #   typepy
pyyaml==6.0.2
    # via
    #   accelerate
    #   awscli
    #   datamodel-code-generator
    #   datasets
540
    #   genai-perf
541
542
543
    #   huggingface-hub
    #   peft
    #   ray
544
    #   responses
545
    #   schemathesis
546
547
    #   timm
    #   transformers
548
    #   vocos
549
550
rapidfuzz==3.12.1
    # via jiwer
Rui Qiao's avatar
Rui Qiao committed
551
ray==2.43.0
552
    # via -r requirements/test.in
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
redis==5.2.0
    # via tensorizer
referencing==0.35.1
    # via
    #   jsonschema
    #   jsonschema-specifications
regex==2024.9.11
    # via
    #   nltk
    #   sacrebleu
    #   tiktoken
    #   transformers
requests==2.32.3
    # via
    #   buildkite-test-collector
    #   datasets
    #   evaluate
    #   huggingface-hub
    #   lm-eval
572
    #   mistral-common
573
574
    #   pooch
    #   ray
575
    #   responses
576
577
    #   schemathesis
    #   starlette-testclient
578
579
    #   tiktoken
    #   transformers
580
581
responses==0.25.3
    # via genai-perf
582
583
584
585
rfc3339-validator==0.1.4
    # via jsonschema
rfc3987==1.3.8
    # via jsonschema
586
rich==13.9.4
587
588
589
    # via
    #   genai-perf
    #   typer
590
591
rouge-score==0.1.2
    # via lm-eval
592
rpds-py==0.20.1
593
594
595
596
597
    # via
    #   jsonschema
    #   referencing
rsa==4.7.2
    # via awscli
598
runai-model-streamer==0.11.0
599
    # via -r requirements/test.in
600
runai-model-streamer-s3==0.11.0
601
    # via -r requirements/test.in
602
603
604
605
606
607
608
609
610
611
612
613
s3transfer==0.10.3
    # via
    #   awscli
    #   boto3
sacrebleu==2.4.3
    # via lm-eval
safetensors==0.4.5
    # via
    #   accelerate
    #   peft
    #   timm
    #   transformers
614
615
schemathesis==3.39.15
    # via -r requirements/test.in
616
617
618
619
620
621
622
623
624
625
scikit-learn==1.5.2
    # via
    #   librosa
    #   lm-eval
    #   sentence-transformers
scipy==1.13.1
    # via
    #   librosa
    #   scikit-learn
    #   sentence-transformers
626
    #   statsmodels
627
    #   vocos
628
sentence-transformers==3.2.1
629
    # via -r requirements/test.in
630
631
sentencepiece==0.2.0
    # via mistral-common
632
633
setuptools==75.8.0
    # via
Shinichi Hemmi's avatar
Shinichi Hemmi committed
634
    #   mamba-ssm
635
636
    #   pytablewriter
    #   torch
637
638
shellingham==1.5.4
    # via typer
639
640
six==1.16.0
    # via
641
    #   junit-xml
642
    #   python-dateutil
643
    #   rfc3339-validator
644
645
646
647
648
    #   rouge-score
sniffio==1.3.1
    # via
    #   anyio
    #   httpx
649
650
sortedcontainers==2.4.0
    # via hypothesis
651
652
soundfile==0.12.1
    # via
653
    #   -r requirements/test.in
654
655
656
657
658
    #   librosa
soxr==0.5.0.post1
    # via librosa
sqlitedict==2.1.0
    # via lm-eval
659
660
661
662
663
664
starlette==0.46.2
    # via
    #   schemathesis
    #   starlette-testclient
starlette-testclient==0.4.1
    # via schemathesis
665
666
statsmodels==0.14.4
    # via genai-perf
667
sympy==1.13.1
668
669
670
    # via
    #   einx
    #   torch
671
672
673
674
675
676
677
tabledata==1.3.3
    # via pytablewriter
tabulate==0.9.0
    # via sacrebleu
tcolorpy==0.1.6
    # via pytablewriter
tenacity==9.0.0
678
679
680
    # via
    #   lm-eval
    #   plotly
681
tensorizer==2.9.0
682
    # via -r requirements/test.in
683
684
threadpoolctl==3.5.0
    # via scikit-learn
685
686
687
688
tiktoken==0.7.0
    # via
    #   lm-eval
    #   mistral-common
689
timm==1.0.11
690
    # via -r requirements/test.in
691
692
693
694
tokenizers==0.21.1
    # via
    #   -r requirements/test.in
    #   transformers
695
696
697
698
tomli==2.2.1
    # via schemathesis
tomli-w==1.2.0
    # via schemathesis
Michael Goin's avatar
Michael Goin committed
699
torch==2.6.0
700
    # via
701
    #   -r requirements/test.in
702
703
    #   accelerate
    #   bitsandbytes
704
    #   encodec
705
    #   fastsafetensors
706
    #   lm-eval
Shinichi Hemmi's avatar
Shinichi Hemmi committed
707
    #   mamba-ssm
708
    #   peft
709
    #   runai-model-streamer
710
711
712
    #   sentence-transformers
    #   tensorizer
    #   timm
713
    #   torchaudio
714
    #   torchvision
715
716
    #   vector-quantize-pytorch
    #   vocos
Michael Goin's avatar
Michael Goin committed
717
torchaudio==2.6.0
718
    # via
719
    #   -r requirements/test.in
720
721
    #   encodec
    #   vocos
Michael Goin's avatar
Michael Goin committed
722
723
724
725
torchvision==0.21.0
    # via
    #   -r requirements/test.in
    #   timm
726
727
728
729
730
731
732
733
tqdm==4.66.6
    # via
    #   datasets
    #   evaluate
    #   huggingface-hub
    #   lm-eval
    #   nltk
    #   peft
734
    #   pqdm
735
736
737
738
739
    #   sentence-transformers
    #   tqdm-multiprocess
    #   transformers
tqdm-multiprocess==0.0.11
    # via lm-eval
740
transformers==4.51.1
741
    # via
742
    #   -r requirements/test.in
743
    #   genai-perf
744
    #   lm-eval
Shinichi Hemmi's avatar
Shinichi Hemmi committed
745
    #   mamba-ssm
746
747
748
749
    #   peft
    #   sentence-transformers
    #   transformers-stream-generator
transformers-stream-generator==0.0.5
750
    # via -r requirements/test.in
Michael Goin's avatar
Michael Goin committed
751
triton==3.2.0
752
    # via torch
753
754
tritonclient==2.51.0
    # via
755
    #   -r requirements/test.in
756
    #   genai-perf
757
typepy==1.3.2
758
759
760
761
    # via
    #   dataproperty
    #   pytablewriter
    #   tabledata
762
763
typer==0.15.2
    # via fastsafetensors
764
765
types-python-dateutil==2.9.0.20241206
    # via arrow
766
767
768
769
typing-extensions==4.12.2
    # via
    #   huggingface-hub
    #   librosa
770
    #   mistral-common
771
    #   pqdm
772
773
774
    #   pydantic
    #   pydantic-core
    #   torch
775
    #   typer
776
777
tzdata==2024.2
    # via pandas
778
779
uri-template==1.3.0
    # via jsonschema
780
urllib3==2.2.3
781
    # via
782
    #   blobfile
783
784
    #   botocore
    #   requests
785
786
    #   responses
    #   tritonclient
787
vector-quantize-pytorch==1.21.2
788
    # via -r requirements/test.in
789
vocos==0.1.0
790
    # via -r requirements/test.in
791
792
793
794
webcolors==24.11.1
    # via jsonschema
werkzeug==3.1.3
    # via schemathesis
795
796
797
798
799
800
word2number==1.1
    # via lm-eval
xxhash==3.5.0
    # via
    #   datasets
    #   evaluate
801
yarl==1.17.1
802
803
804
    # via
    #   aiohttp
    #   schemathesis
805
806
zstandard==0.23.0
    # via lm-eval