torch>=2.1
pytorch-lightning<3.0,>=2.1
torchvision
numpy
kornia
wandb
omegaconf
matplotlib
rasterio
scikit-image
Pillow
huggingface-hub

[docs]
mkdocs-material
mkdocstrings[python]
pymdown-extensions

[hpc]
PyYAML>=6.0
pyproj>=3.4
pyshp>=2.3.1
rioxarray>=0.15
shapely>=2.0.2
cubo>=0.3
opensr-utils>=1.0.0

[tests]
pytest
pytest-cov
numpy
