LICENSE
README.md
pyproject.toml
tests/test_interface.py
tests/test_metric.py
tests/test_miniwell.py
the_well/__init__.py
the_well.egg-info/PKG-INFO
the_well.egg-info/SOURCES.txt
the_well.egg-info/dependency_links.txt
the_well.egg-info/entry_points.txt
the_well.egg-info/requires.txt
the_well.egg-info/top_level.txt
the_well/benchmark/__init__.py
the_well/benchmark/train.py
the_well/benchmark/metrics/__init__.py
the_well/benchmark/metrics/common.py
the_well/benchmark/metrics/plottable_data.py
the_well/benchmark/metrics/spatial.py
the_well/benchmark/metrics/spectral.py
the_well/benchmark/metrics/temporal.py
the_well/benchmark/models/__init__.py
the_well/benchmark/models/common.py
the_well/benchmark/models/afno/__init__.py
the_well/benchmark/models/avit/__init__.py
the_well/benchmark/models/dilated_resnet/__init__.py
the_well/benchmark/models/fno/__init__.py
the_well/benchmark/models/refno/__init__.py
the_well/benchmark/models/tfno/__init__.py
the_well/benchmark/models/unet_classic/__init__.py
the_well/benchmark/models/unet_convnext/__init__.py
the_well/benchmark/optim/schedulers.py
the_well/benchmark/trainer/__init__.py
the_well/benchmark/trainer/training.py
the_well/benchmark/trainer/utils.py
the_well/benchmark/utils/__init__.py
the_well/benchmark/utils/experiment_utils.py
the_well/data/__init__.py
the_well/data/augmentation.py
the_well/data/data_formatter.py
the_well/data/datamodule.py
the_well/data/datasets.py
the_well/data/miniwell.py
the_well/data/normalization.py
the_well/data/utils.py
the_well/utils/__init__.py
the_well/utils/download.py
the_well/utils/dummy_data.py
the_well/utils/export.py
the_well/utils/registry.yaml
the_well/utils/utils.py
the_well/utils/interface/__init__.py