pyproject.toml

[project]
name = "aifs-ens"
version = "2.0"
requires-python = ">=3.11,<3.14"
classifiers = [
  "Programming Language :: Python :: 3 :: Only",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
]
dependencies = [
  "anemoi-graphs==0.8.1",
  "anemoi-models==0.11.2",
  "anemoi-transform==0.1.16.post2",
  "anemoi-utils==0.4.35.post3",
  "earthkit-data<1",
  "earthkit-plots<1",
  "flash-attn==2.7.4",
  "torch==2.7",
  "torch-geometric==2.6.1",
]
optional-dependencies.inference = [
  "anemoi-inference[huggingface]==0.8.3",
  "anemoi-plugins-ecmwf-inference[opendata]==0.2.1",
  "earthkit-regrid==0.5.1",
  "ecmwf-opendata==0.3.29",
]
optional-dependencies.jupyter = [
  "ipykernel>=7.2",
]
#optional-dependencies.training = [
#  "anemoi-training==0.8.1",
#]

[tool.uv]
sources.flash-attn = [
  { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl", marker = "platform_machine == 'x86_64' and sys_platform != 'win32'" },
  { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl", marker = "platform_machine == 'aarch64' and sys_platform == 'linux'" },
]
environments = [
  "sys_platform == 'linux' and (platform_machine == 'x86_64' or platform_machine == 'aarch64')",
]