torch>2.0
torchvision
rioxarray
albumentations==1.4.10
albucore==0.0.16
rasterio
torchmetrics
geopandas
lightly
jsonargparse<=4.35.0
h5py
lightning
segmentation-models-pytorch==0.5.0
jsonargparse
torchgeo<0.7.2,>=0.7.0
einops
timm>=1.0.15
pycocotools
huggingface_hub
tifffile
python-box
tqdm
wandb
tensorboard

[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<=0.34.0
tokenizers
dask
wandb
pytest-sugar
numba
hdf5plugin
h5netcdf

[visualize]
torchview
graphviz

[vllm]
geobenchv2==0.9
vllm==0.11.0

[vllm_test]
pillow
imagehash

[wxc]
PrithviWxC
granitewxc
