-
Notifications
You must be signed in to change notification settings - Fork 0
/
constants.py
32 lines (24 loc) · 886 Bytes
/
constants.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
from weather_weaver.config import DATA_DIR
# Data source to download ECMWF open data from
DATA_SOURCE = "azure"
# list of parameters to download limited by what is available in the open-data catalogue
# read more about request parameters https://github.com/ecmwf/ecmwf-opendata?tab=readme-ov-file#parameters-and-levels
NWP_PARAMETERS = [
"2t",
"tp",
"10u",
"10v",
]
# limit to 90 hourly steps by default
FORECAST_STEPS = [3 * i for i in range(31)]
# Output directories
ECMWF_DIR = DATA_DIR / "ecmwf"
ECMWF_DIR.mkdir(exist_ok=True)
RAW_DIR = ECMWF_DIR / "raw"
RAW_DIR.mkdir(exist_ok=True)
PROCESSED_DIR = ECMWF_DIR / "processed"
PROCESSED_DIR.mkdir(exist_ok=True)
# list of coordinates to keep from raw datasets
COORDINATE_ALLOW_LIST: list[str] = ["time", "step", "latitude", "longitude"]
# The target resolution when resampling datasets.
TARGET_RESOLUTION = 0.2