requirements.txt

spaces

torch==2.11.0
torch-geometric==2.6.1
earthkit-regrid
ecmwf-opendata
# Pinned <1 deliberately. anemoi-transform/anemoi-inference already require
# earthkit-data<1 transitively, so this only makes the existing constraint
# explicit — but it matters: 1.x dropped source subscripting, which several
# call sites here rely on (see aifs.initial_conditions._ocean_mask). Raising
# this bound means auditing every ekd.from_source(...) use first.
earthkit-data<1

anemoi-models==0.9.3
anemoi-transform==0.4.2
anemoi-datasets==0.5.26
anemoi-graphs==0.6.4
anemoi-inference

gradio
matplotlib
cartopy

# WeatherNext 2 (Google DeepMind) support — pinned to an unmerged transformers
# fork since WeatherNext2ForWeatherForecasting/WeatherNext2FeatureExtractor
# aren't in an official transformers release yet. This is inherently fragile
# (branch could change/disappear) — swap for a release tag once the PR merges.
git+https://github.com/kashif/transformers.git@add-weathernext2
accelerate

# Forecast archive (save/load runs to the EmmaScharfmann/weather-forecast-archive
# HF dataset) — already an implicit dependency of transformers, listed
# explicitly since aifs/archive.py imports it directly.
huggingface_hub

# HF Space environment config:
# Python: 3.12.12 (main, Feb 24 2026, 21:49:09) [GCC 14.2.0]
#Platform: Linux-6.12.80-106.156.amzn2023.x86_64-x86_64-with-glibc2.41
#Torch: 2.11.0+cu130
#CUDA: 13.0
#CXX11 ABI: True
#GPU: NVIDIA RTX PRO 6000 Blackwell Server Edition

# pip install \
#  "anemoi-models==0.9.3" \
#  "anemoi-transform==0.4.2"

#pip install \
#  "anemoi-datasets==0.5.26" \
#  "anemoi-graphs==0.6.4"

# pip install anemoi-inference