albumentations<2,>=1.3
click<9,>=8.1
geopandas<2,>=0.14
numpy<2,>=1.26
pytorch-lightning<3,>=2.1
rasterio<2,>=1.3
scikit-learn<2,>=1.3
segmentation-models-pytorch==0.5.0
shapely<3,>=2.0
torch<3,>=2.1
torchvision<1,>=0.16
torchmetrics<2,>=1.7
rich<14,>=13.0

[dev]
pytest>=7.0
black>=23.0
flake8>=6.0
