LICENSE
README.md
pyproject.toml
src/climate_learn/__init__.py
src/climate_learn.egg-info/PKG-INFO
src/climate_learn.egg-info/SOURCES.txt
src/climate_learn.egg-info/dependency_links.txt
src/climate_learn.egg-info/requires.txt
src/climate_learn.egg-info/top_level.txt
src/climate_learn/data/__init__.py
src/climate_learn/data/climatebench_dataset.py
src/climate_learn/data/climatebench_module.py
src/climate_learn/data/download.py
src/climate_learn/data/iterdataset.py
src/climate_learn/data/itermodule.py
src/climate_learn/data/mapmodule.py
src/climate_learn/data/module.py
src/climate_learn/data/npzdataset.py
src/climate_learn/data/climate_dataset/__init__.py
src/climate_learn/data/climate_dataset/climate_dataset.py
src/climate_learn/data/climate_dataset/cmip6_iterdataset.py
src/climate_learn/data/climate_dataset/era5_continuous_iterdataset.py
src/climate_learn/data/climate_dataset/era5_iterdataset.py
src/climate_learn/data/climate_dataset/stacked_climate_dataset.py
src/climate_learn/data/climate_dataset/args/__init__.py
src/climate_learn/data/climate_dataset/args/climate_dataset_args.py
src/climate_learn/data/climate_dataset/args/era5_args.py
src/climate_learn/data/climate_dataset/args/stacked_climate_dataset_args.py
src/climate_learn/data/climate_dataset/cmip6/constants.py
src/climate_learn/data/climate_dataset/era5/__init__.py
src/climate_learn/data/climate_dataset/era5/constants.py
src/climate_learn/data/climate_dataset/era5/era5.py
src/climate_learn/data/dataset/__init__.py
src/climate_learn/data/dataset/map_dataset.py
src/climate_learn/data/dataset/shard_dataset.py
src/climate_learn/data/dataset/args/__init__.py
src/climate_learn/data/dataset/args/map_dataset_args.py
src/climate_learn/data/dataset/args/shard_dataset_args.py
src/climate_learn/data/processing/climatebench.py
src/climate_learn/data/processing/cmip6_constants.py
src/climate_learn/data/processing/era5_constants.py
src/climate_learn/data/processing/era5_cropped.py
src/climate_learn/data/processing/era5_extreme.py
src/climate_learn/data/processing/nc2npz.py
src/climate_learn/data/processing/prism.py
src/climate_learn/data/task/__init__.py
src/climate_learn/data/task/downscaling.py
src/climate_learn/data/task/forecasting.py
src/climate_learn/data/task/task.py
src/climate_learn/data/task/args/__init__.py
src/climate_learn/data/task/args/downscaling_args.py
src/climate_learn/data/task/args/forecasting_args.py
src/climate_learn/data/task/args/task_args.py
src/climate_learn/metrics/__init__.py
src/climate_learn/metrics/functional.py
src/climate_learn/metrics/metrics.py
src/climate_learn/metrics/utils.py
src/climate_learn/models/__init__.py
src/climate_learn/models/lr_scheduler.py
src/climate_learn/models/module.py
src/climate_learn/models/hub/__init__.py
src/climate_learn/models/hub/climatology.py
src/climate_learn/models/hub/interpolation.py
src/climate_learn/models/hub/linear_regression.py
src/climate_learn/models/hub/persistence.py
src/climate_learn/models/hub/resnet.py
src/climate_learn/models/hub/unet.py
src/climate_learn/models/hub/utils.py
src/climate_learn/models/hub/vit.py
src/climate_learn/models/hub/components/cnn_blocks.py
src/climate_learn/models/hub/components/pos_embed.py
src/climate_learn/transforms/__init__.py
src/climate_learn/transforms/denormalize.py
src/climate_learn/transforms/mask.py
src/climate_learn/transforms/registry.py
src/climate_learn/utils/__init__.py
src/climate_learn/utils/loaders.py
src/climate_learn/utils/mc_dropout.py
src/climate_learn/utils/visualize.py
tests/test_trainer.py