torch==2.4.0
numpy==1.26.0
segmentation_models_pytorch==0.3.3
torchvision>=0.19
torchmetrics>=1.6
xarray==2022.3.0
rasterio==1.4a3
rioxarray==0.17.0
shapely==2.0.5
scipy==1.14.0
scikit-image==0.24.0
hyperopt==0.2.7
fiona==1.9.6
psutil==6.0.0
geojson==3.1.0
geoplot==0.5.1
folium==0.17.0
hydra-core==1.3
lightning==2.4.0
onnx==1.16.2
rootutils>=1.0
rich>=13.0
mlflow==2.18.0
ipykernel
jsonschema2md>=1.3

[test]
pytest
