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