torch>2.0
numpy>=2.2
torchvision
rioxarray
albumentations
albucore
rasterio
torchmetrics
geopandas
lightly==1.5.22
h5py
lightning>=2.6.0
segmentation-models-pytorch>=0.5.0
jsonargparse>=4.40.0
torchgeo
einops
timm>=1.0.15
pycocotools
huggingface_hub
tifffile
python-box
tqdm
wandb>=0.18.0
tacoreader==0.5.6
tensorboard
diffusers
scikit-learn>=1.3.2
scikit-image
pyarrow
rich
termcolor

[dev]
black
mkdocs-material
mkdocstrings[python]
mike
mkdocs-git-revision-date-localized-plugin
pre-commit

[geobenchv2]
geobenchv2==0.9
mlflow

[mmseg]
mmcv
ftfy
regex
openmim

[peft]
peft>=0.15.0

[surya]
numba
hdf5plugin
h5netcdf
terratorch-surya==0.1.0

[test]
pytest
pytest-coverage
coverage-badge
peft>=0.15.0
diffusers
tokenizers
dask
numba
hdf5plugin
h5netcdf
geobenchv2>=0.9
terratorch-surya>=0.1.0

[visualize]
torchview
graphviz

[vllm]
geobenchv2>=0.9
impactmesh>0.1.0
vllm!=0.15.*,>=0.12

[vllm_test]
pillow
imagehash

[wxc]
PrithviWxC
granitewxc
