Download NWM Retrospective Gridded Data

Import the TEEHR module and any required libraries.

from pathlib import Path

from teehr.loading.nwm.retrospective_grids import nwm_retro_grids_to_parquet

Specify the input parameters.

OUTPUT_ROOT = Path(Path().home(), "temp")
OUTPUT_DIR = Path(OUTPUT_ROOT, "gridded_retro")

ZONAL_WEIGHTS = Path(
    "tests", "data", "nwm22", "onehuc10_weights_retro.parquet"
)

NWM_VERSION = "nwm30"
VARIABLE_NAME = "RAINRATE"
START_DATE = "2008-05-23 09:00"
END_DATE = "2008-05-23 10:00"
OVERWRITE = True
CHUNK_BY = None
LOCATION_ID_PREFIX = "ngen"

Fetch and load the data.

nwm_retro_grids_to_parquet(
    nwm_version=NWM_VERSION,
    variable_name=VARIABLE_NAME,
    zonal_weights_filepath=ZONAL_WEIGHTS,
    start_date=START_DATE,
    end_date=END_DATE,
    output_parquet_dir=OUTPUT_DIR,
    overwrite_output=OVERWRITE,
    chunk_by=CHUNK_BY,
    location_id_prefix=LOCATION_ID_PREFIX
)