| [project] | |
| name = "coarse-grained-forecasting" | |
| version = "1.0.0" | |
| description = "Direct SI, Cascaded SI and EDM for conditional forecasting from coarse-grained observations: the code of the paper's models, training, sampling and csFID evaluation." | |
| readme = "README.md" | |
| license = { file = "LICENSE" } | |
| requires-python = ">=3.10,<3.11" | |
| dependencies = [ | |
| "torch==2.1.2", | |
| "torchvision==0.16.2", | |
| "numpy==1.26.2", | |
| "scipy==1.15.3", | |
| "pyyaml==6.0.1", | |
| "pandas==2.2.1", | |
| "matplotlib==3.8.2", | |
| "scikit-learn==1.3.2", | |
| "h5py==3.15.1", | |
| "einops==0.8.1", | |
| "Pillow==10.0.1", | |
| "clean-fid==0.1.35", | |
| "requests", | |
| ] | |
| [tool.uv] | |
| # PyTorch 2.1.2 built for CUDA 11.8 (the environment of the recorded Euler, NLKG and passive-scalar values); | |
| # on a machine without CUDA the same wheels run on the CPU. | |
| [[tool.uv.index]] | |
| name = "pytorch-cu118" | |
| url = "https://download.pytorch.org/whl/cu118" | |
| explicit = true | |
| [tool.uv.sources] | |
| torch = { index = "pytorch-cu118" } | |
| torchvision = { index = "pytorch-cu118" } | |
| [build-system] | |
| requires = ["setuptools>=61"] | |
| build-backend = "setuptools.build_meta" | |
| [tool.setuptools.packages.find] | |
| include = ["bifurcation*", "reproduce*"] | |