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