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