[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*"]