Note
Go to the end to download the full example code.
Running StormCast Inference#
Basic StormCast inference workflow.
This example will demonstrate how to run a simple inference workflow to generate a basic determinstic forecast using StormCast. For details about the stormcast model, see
# /// script
# dependencies = [
# "earth2studio[data,stormcast] @ git+https://github.com/NVIDIA/earth2studio.git",
# "cartopy",
# ]
# ///
Set Up#
All workflows inside Earth2Studio require constructed components to be
handed to them. In this example, let’s take a look at the most basic:
earth2studio.run.deterministic().
Thus, we need the following:
Prognostic Model: Use the built in StormCast Model
earth2studio.models.px.StormCast.Datasource: Pull data from the HRRR data api
earth2studio.data.HRRR.IO Backend: Let’s save the outputs into a Zarr store
earth2studio.io.ZarrBackend.
StormCast also requires a conditioning data source. We use a forecast data source here,
GFS_FX earth2studio.data.GFS_FX which is the default, but a non-forecast
data source such as ARCO could also be used with appropriate time stamps.
from datetime import datetime, timedelta
from loguru import logger
from tqdm import tqdm
logger.remove()
logger.add(lambda msg: tqdm.write(msg, end=""), colorize=True)
import os
os.makedirs("outputs", exist_ok=True)
from dotenv import load_dotenv
load_dotenv() # TODO: make common example prep function
from earth2studio.data import HRRR
from earth2studio.io import ZarrBackend
from earth2studio.models.px import StormCast
# Load the default model package which downloads the check point from NGC
# Use the default conditioning data source GFS_FX
package = StormCast.load_default_package()
model = StormCast.load_model(package)
# Create the data source
data = HRRR()
# Create the IO handler, store in memory
io = ZarrBackend()
Execute the Workflow#
With all components initialized, running the workflow is a single line of Python code. Workflow will return the provided IO object back to the user, which can be used to then post process. Some have additional APIs that can be handy for post-processing or saving to file. Check the API docs for more information.
For the forecast we will predict for 4 hours
import earth2studio.run as run
nsteps = 4
today = datetime.today() - timedelta(days=1)
date = today.isoformat().split("T")[0]
io = run.deterministic([date], nsteps, model, data, io)
print(io.root.tree())
2026-03-24 22:16:17.115 | INFO | earth2studio.run:deterministic:78 - Running simple workflow!
2026-03-24 22:16:17.115 | INFO | earth2studio.run:deterministic:85 - Inference device: cuda
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.717 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 189981799-925306
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.720 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 60024203-1233115
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.722 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 65540416-2187742
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.724 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 141136864-1185672
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.726 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 88457893-945033
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.728 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 114691302-1286114
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.729 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 29670238-2369842
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.731 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 113675793-1015509
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.733 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 187608751-2373048
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.735 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 143261289-945348
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.737 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 0-2188827
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.738 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 86102037-1056306
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.740 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 168711705-885383
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.741 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 127166073-993744
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.743 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 33126552-1220046
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.745 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 239499008-2168448
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.747 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 120267241-2145907
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.748 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 46577536-1213122
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.750 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfsfcf00.grib2 41806060-2143472
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.752 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 7152532-1287445
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.754 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 245209926-791451
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.755 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 48810908-1024728
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.757 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 102306019-950409
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.759 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 74751819-955201
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.760 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 247758382-730071
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.762 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 247014153-744229
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.763 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 99979713-1035716
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.764 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 124732596-2433477
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.765 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 103256428-980837
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.767 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 295556944-637512
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.768 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 142322536-938753
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.769 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 129411767-954444
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.770 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 20149643-1248444
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.771 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfsfcf00.grib2 0-180657
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.772 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 115977416-947359
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.773 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 193294993-834538
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.774 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 97549282-2430431
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.775 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 45494317-1083219
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.776 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 340574252-1305732
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.777 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 192438074-856919
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.779 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 140166875-969989
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.780 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 79147179-2185938
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.781 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 6031080-1121452
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.782 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 25856022-2187936
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.782 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 190907105-1530969
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.783 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 106500196-2171495
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.784 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 19057732-1091911
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.784 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 344045631-762112
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.785 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 296194456-1287218
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.785 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 158467725-2295039
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.786 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 8439977-1050687
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.787 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 342488833-1556798
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.787 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 128159817-1251950
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.788 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 34346598-1044475
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.789 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 130366211-967108
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.789 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 12900274-2188321
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.790 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 298202161-714395
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.790 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 52135877-2187301
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.791 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 21398087-1055594
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.792 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 293752964-1803980
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.792 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 58943942-1080261
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.793 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 56547995-2395947
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.794 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfsfcf00.grib2 43949532-2381615
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.794 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 101015429-1290590
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.795 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 47790658-1020250
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.796 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 22453681-1066898
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.797 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 137735830-2431045
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.797 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 72390758-1075297
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.798 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 62244222-1001210
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.799 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 165120856-956661
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.799 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 75707020-985012
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.800 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 243079898-2130028
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.801 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 83678711-2423326
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.801 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 167817424-894281
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.802 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 3760751-2270329
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.802 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 111242778-2433015
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.803 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 32040080-1086472
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.804 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 16703717-2354015
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.804 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfsfcf00.grib2 34633080-1218180
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.805 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 9490664-1074698
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.806 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 38761475-2188467
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.806 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 297481674-720487
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.807 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 246001377-1012776
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.807 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfsfcf00.grib2 23710598-618225
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.808 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 344807743-736975
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.809 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 133449024-2114722
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.809 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 89402926-978662
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.810 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 61257318-986904
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.811 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 92826527-2180582
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.811 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 166077517-1739907
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.812 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 162708655-2412201
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.812 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 73466055-1285764
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.813 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 43114149-2380168
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.814 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 116924775-978628
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.814 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 183638781-2224931
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.815 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 35391073-1046221
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.816 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 69979689-2411069
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.816 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 87158343-1299550
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2026-03-24 22:16:22.817 | DEBUG | earth2studio.data.hrrr:fetch_array:509 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20260323/conus/hrrr.t00z.wrfnatf00.grib2 341879984-608849
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
Fetching HRRR data: 1%| | 1/99 [00:00<01:12, 1.35it/s]
Fetching HRRR data: 2%|▏ | 2/99 [00:00<00:40, 2.37it/s]
Fetching HRRR data: 3%|▎ | 3/99 [00:01<00:29, 3.28it/s]
Fetching HRRR data: 4%|▍ | 4/99 [00:01<00:25, 3.72it/s]
Fetching HRRR data: 5%|▌ | 5/99 [00:01<00:19, 4.76it/s]
Fetching HRRR data: 7%|▋ | 7/99 [00:01<00:14, 6.43it/s]
Fetching HRRR data: 8%|▊ | 8/99 [00:01<00:14, 6.45it/s]
Fetching HRRR data: 9%|▉ | 9/99 [00:01<00:14, 6.20it/s]
Fetching HRRR data: 11%|█ | 11/99 [00:02<00:12, 6.80it/s]
Fetching HRRR data: 12%|█▏ | 12/99 [00:02<00:12, 6.91it/s]
Fetching HRRR data: 13%|█▎ | 13/99 [00:02<00:12, 7.09it/s]
Fetching HRRR data: 15%|█▌ | 15/99 [00:02<00:09, 9.14it/s]
Fetching HRRR data: 16%|█▌ | 16/99 [00:02<00:09, 8.42it/s]
Fetching HRRR data: 18%|█▊ | 18/99 [00:03<00:09, 8.37it/s]
Fetching HRRR data: 19%|█▉ | 19/99 [00:03<00:09, 8.38it/s]
Fetching HRRR data: 21%|██ | 21/99 [00:03<00:08, 9.26it/s]
Fetching HRRR data: 24%|██▍ | 24/99 [00:03<00:07, 10.22it/s]
Fetching HRRR data: 26%|██▋ | 26/99 [00:03<00:06, 10.81it/s]
Fetching HRRR data: 28%|██▊ | 28/99 [00:04<00:08, 8.57it/s]
Fetching HRRR data: 31%|███▏ | 31/99 [00:04<00:06, 10.58it/s]
Fetching HRRR data: 33%|███▎ | 33/99 [00:04<00:06, 9.62it/s]
Fetching HRRR data: 35%|███▌ | 35/99 [00:04<00:06, 10.14it/s]
Fetching HRRR data: 37%|███▋ | 37/99 [00:05<00:07, 7.99it/s]
Fetching HRRR data: 38%|███▊ | 38/99 [00:05<00:07, 8.15it/s]
Fetching HRRR data: 40%|████ | 40/99 [00:05<00:06, 9.02it/s]
Fetching HRRR data: 41%|████▏ | 41/99 [00:05<00:07, 7.47it/s]
Fetching HRRR data: 43%|████▎ | 43/99 [00:05<00:06, 8.45it/s]
Fetching HRRR data: 45%|████▌ | 45/99 [00:05<00:05, 10.26it/s]
Fetching HRRR data: 47%|████▋ | 47/99 [00:06<00:05, 10.27it/s]
Fetching HRRR data: 49%|████▉ | 49/99 [00:06<00:07, 6.86it/s]
Fetching HRRR data: 52%|█████▏ | 51/99 [00:06<00:07, 6.15it/s]
Fetching HRRR data: 53%|█████▎ | 52/99 [00:07<00:07, 6.24it/s]
Fetching HRRR data: 54%|█████▎ | 53/99 [00:07<00:08, 5.12it/s]
Fetching HRRR data: 55%|█████▍ | 54/99 [00:10<00:35, 1.27it/s]
Fetching HRRR data: 56%|█████▌ | 55/99 [00:10<00:30, 1.44it/s]
Fetching HRRR data: 57%|█████▋ | 56/99 [00:10<00:25, 1.72it/s]
Fetching HRRR data: 58%|█████▊ | 57/99 [00:11<00:19, 2.12it/s]
Fetching HRRR data: 59%|█████▊ | 58/99 [00:11<00:15, 2.70it/s]
Fetching HRRR data: 60%|█████▉ | 59/99 [00:11<00:13, 3.01it/s]
Fetching HRRR data: 61%|██████ | 60/99 [00:11<00:12, 3.09it/s]
Fetching HRRR data: 62%|██████▏ | 61/99 [00:11<00:10, 3.78it/s]
Fetching HRRR data: 63%|██████▎ | 62/99 [00:11<00:08, 4.59it/s]
Fetching HRRR data: 64%|██████▎ | 63/99 [00:12<00:08, 4.06it/s]
Fetching HRRR data: 65%|██████▍ | 64/99 [00:12<00:08, 3.95it/s]
Fetching HRRR data: 66%|██████▌ | 65/99 [00:12<00:07, 4.55it/s]
Fetching HRRR data: 68%|██████▊ | 67/99 [00:12<00:05, 5.82it/s]
Fetching HRRR data: 69%|██████▊ | 68/99 [00:13<00:05, 6.19it/s]
Fetching HRRR data: 70%|██████▉ | 69/99 [00:13<00:05, 5.80it/s]
Fetching HRRR data: 71%|███████ | 70/99 [00:13<00:05, 5.66it/s]
Fetching HRRR data: 73%|███████▎ | 72/99 [00:13<00:04, 6.19it/s]
Fetching HRRR data: 75%|███████▍ | 74/99 [00:13<00:03, 7.42it/s]
Fetching HRRR data: 78%|███████▊ | 77/99 [00:14<00:02, 7.73it/s]
Fetching HRRR data: 80%|███████▉ | 79/99 [00:14<00:02, 9.22it/s]
Fetching HRRR data: 82%|████████▏ | 81/99 [00:15<00:03, 5.49it/s]
Fetching HRRR data: 83%|████████▎ | 82/99 [00:18<00:13, 1.28it/s]
Fetching HRRR data: 84%|████████▍ | 83/99 [00:19<00:11, 1.39it/s]
Fetching HRRR data: 85%|████████▍ | 84/99 [00:19<00:10, 1.38it/s]
Fetching HRRR data: 87%|████████▋ | 86/99 [00:20<00:06, 1.96it/s]
Fetching HRRR data: 88%|████████▊ | 87/99 [00:20<00:05, 2.00it/s]
Fetching HRRR data: 90%|████████▉ | 89/99 [00:21<00:04, 2.44it/s]
Fetching HRRR data: 91%|█████████ | 90/99 [00:21<00:03, 2.85it/s]
Fetching HRRR data: 92%|█████████▏| 91/99 [00:21<00:03, 2.55it/s]
Fetching HRRR data: 93%|█████████▎| 92/99 [00:22<00:03, 1.83it/s]
Fetching HRRR data: 94%|█████████▍| 93/99 [00:23<00:03, 1.78it/s]
Fetching HRRR data: 95%|█████████▍| 94/99 [00:25<00:05, 1.03s/it]
Fetching HRRR data: 96%|█████████▌| 95/99 [00:32<00:10, 2.69s/it]
Fetching HRRR data: 97%|█████████▋| 96/99 [00:34<00:07, 2.45s/it]
Fetching HRRR data: 98%|█████████▊| 97/99 [00:35<00:03, 1.90s/it]
Fetching HRRR data: 99%|█████████▉| 98/99 [00:39<00:02, 2.68s/it]
Fetching HRRR data: 100%|██████████| 99/99 [00:54<00:00, 6.30s/it]
Fetching HRRR data: 100%|██████████| 99/99 [00:54<00:00, 1.81it/s]
2026-03-24 22:17:17.659 | SUCCESS | earth2studio.run:deterministic:109 - Fetched data from HRRR
2026-03-24 22:17:17.716 | INFO | earth2studio.run:deterministic:139 - Inference starting!
Running inference: 0%| | 0/5 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:01<00:04, 1.02s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2026-03-24 22:17:23.618 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 0-1009660
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.621 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 405943665-957905
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.624 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 211150956-734241
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.626 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 399966877-867249
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.628 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 347943697-949074
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.630 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 344314591-1239571
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.633 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 412209310-839132
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.635 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 424380869-974472
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.637 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 217151377-588898
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.639 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 348892771-954998
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.641 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 420780114-514073
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.643 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 271063132-920203
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.645 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 213845804-1164188
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.647 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 410121076-1001673
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.650 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 432190802-1189674
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.652 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 341208776-906189
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.653 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 211885197-745865
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.655 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 404972422-971243
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.658 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 264532176-721976
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.660 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 270099543-963589
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.662 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 401857197-1223225
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.664 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 266498918-1252627
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.666 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 263722716-809460
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.668 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 425355341-946464
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.669 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 217740275-594480
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
2026-03-24 22:17:23.671 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f000 342114965-844133
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:04, 1.02s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:16, 1.56it/s]
Fetching GFS data: 23%|██▎ | 6/26 [00:00<00:01, 10.10it/s]
Fetching GFS data: 35%|███▍ | 9/26 [00:00<00:01, 12.96it/s]
Fetching GFS data: 50%|█████ | 13/26 [00:01<00:00, 18.04it/s]
Fetching GFS data: 69%|██████▉ | 18/26 [00:01<00:00, 24.33it/s]
Fetching GFS data: 85%|████████▍ | 22/26 [00:11<00:03, 1.11it/s]
Fetching GFS data: 88%|████████▊ | 23/26 [00:14<00:03, 1.16s/it]
Fetching GFS data: 100%|██████████| 26/26 [00:17<00:00, 1.06s/it]
Fetching GFS data: 100%|██████████| 26/26 [00:17<00:00, 1.50it/s]
Running inference: 40%|████ | 2/5 [00:30<00:53, 17.96s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2026-03-24 22:17:48.929 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 426840316-969948
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.932 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 272144902-920181
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.934 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 402770993-1222040
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.936 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 405883162-969830
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.938 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 218088103-593372
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.940 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 345449069-1230983
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.942 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 413131804-839122
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.945 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 411045796-1001036
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.947 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 267578490-1256489
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.949 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 349058990-944335
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.951 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 342344667-905364
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.954 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 214074400-1161779
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.956 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 264797082-809658
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.958 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 0-1008982
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.960 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 211382838-734390
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.962 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 271181857-963045
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.964 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 400879803-865868
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.966 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 427810264-950350
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.968 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 217499644-588459
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.970 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 343250031-843439
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.972 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 422216334-514291
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.974 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 406852992-957545
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.976 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 265606740-721741
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.978 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 350003325-949901
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.980 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 212117228-745536
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
2026-03-24 22:17:48.983 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f001 441858403-1189842
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:31<00:53, 17.96s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:11, 2.20it/s]
Fetching GFS data: 8%|▊ | 2/26 [00:00<00:07, 3.39it/s]
Fetching GFS data: 15%|█▌ | 4/26 [00:00<00:03, 6.69it/s]
Fetching GFS data: 38%|███▊ | 10/26 [00:00<00:00, 18.95it/s]
Fetching GFS data: 58%|█████▊ | 15/26 [00:00<00:00, 25.31it/s]
Fetching GFS data: 88%|████████▊ | 23/26 [00:01<00:00, 38.81it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:10<00:00, 2.47it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2026-03-24 22:18:07.365 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 268259068-1257160
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.368 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 428067135-968894
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.371 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 414365059-839932
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.373 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 349465403-943085
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.376 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 342616744-901594
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.378 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 214970967-1165347
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.380 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 443393037-1188891
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.382 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 265482073-809834
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.384 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 0-1047583
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.387 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 212279161-733812
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.389 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 271861852-966248
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.391 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 401832307-864612
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.393 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 218273047-589191
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.395 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 343518338-847940
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.397 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 407799548-957689
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.399 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 266291907-721679
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.401 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 350408488-952495
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.404 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 213012973-745511
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.406 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 429036029-950251
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.408 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 272828100-920590
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.410 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 423452469-514170
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.413 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 403724224-1221540
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.415 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 406830778-968770
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.417 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 218862238-593432
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.419 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 345724611-1359961
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
2026-03-24 22:18:07.421 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f002 412277028-1000828
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:49<00:36, 18.14s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:11, 2.11it/s]
Fetching GFS data: 8%|▊ | 2/26 [00:00<00:07, 3.29it/s]
Fetching GFS data: 35%|███▍ | 9/26 [00:00<00:01, 16.98it/s]
Fetching GFS data: 62%|██████▏ | 16/26 [00:00<00:00, 26.50it/s]
Fetching GFS data: 85%|████████▍ | 22/26 [00:01<00:00, 31.69it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:05<00:00, 3.06it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:05<00:00, 4.78it/s]
Running inference: 80%|████████ | 4/5 [01:02<00:16, 16.26s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2026-03-24 22:18:21.375 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 273838292-962763
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.378 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 404012991-863664
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.380 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 219829268-588341
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.382 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 345699883-844912
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.384 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 410277355-957342
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.386 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 268268392-721053
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.389 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 352597415-951477
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.391 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 214439807-746044
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.393 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 409307644-969711
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.395 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 274801055-920629
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.397 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 405905510-1221447
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.399 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 431499176-949877
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.402 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 416826144-839996
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.404 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 220417609-593589
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.406 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 347908452-1364285
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.408 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 414741454-997044
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.410 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 270237207-1258164
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.412 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 430529746-969430
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.414 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 351654877-942538
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.416 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 344798526-901357
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.418 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 216400860-1161465
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.420 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 446028073-1189143
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.422 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 267458971-809421
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.424 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 0-1047312
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.426 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 213706073-733734
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
2026-03-24 22:18:21.427 | DEBUG | earth2studio.data.gfs:fetch_array:386 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20260323/00/atmos/gfs.t00z.pgrb2.0p25.f003 425916997-514108
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [01:03<00:16, 16.26s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:16, 1.49it/s]
Fetching GFS data: 12%|█▏ | 3/26 [00:00<00:04, 4.70it/s]
Fetching GFS data: 23%|██▎ | 6/26 [00:00<00:02, 9.04it/s]
Fetching GFS data: 31%|███ | 8/26 [00:01<00:01, 11.29it/s]
Fetching GFS data: 46%|████▌ | 12/26 [00:01<00:00, 15.20it/s]
Fetching GFS data: 58%|█████▊ | 15/26 [00:01<00:00, 18.01it/s]
Fetching GFS data: 73%|███████▎ | 19/26 [00:01<00:00, 23.07it/s]
Fetching GFS data: 85%|████████▍ | 22/26 [00:06<00:02, 1.88it/s]
Fetching GFS data: 92%|█████████▏| 24/26 [00:09<00:01, 1.23it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:11<00:00, 1.25it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:11<00:00, 2.26it/s]
Running inference: 100%|██████████| 5/5 [01:22<00:00, 17.47s/it]
Running inference: 100%|██████████| 5/5 [01:22<00:00, 16.44s/it]
2026-03-24 22:18:39.897 | SUCCESS | earth2studio.run:deterministic:151 -
Inference complete
/
├── Z10hl (1, 5, 512, 640) float32
├── Z11hl (1, 5, 512, 640) float32
├── Z13hl (1, 5, 512, 640) float32
├── Z15hl (1, 5, 512, 640) float32
├── Z1hl (1, 5, 512, 640) float32
├── Z20hl (1, 5, 512, 640) float32
├── Z25hl (1, 5, 512, 640) float32
├── Z2hl (1, 5, 512, 640) float32
├── Z30hl (1, 5, 512, 640) float32
├── Z3hl (1, 5, 512, 640) float32
├── Z4hl (1, 5, 512, 640) float32
├── Z5hl (1, 5, 512, 640) float32
├── Z6hl (1, 5, 512, 640) float32
├── Z7hl (1, 5, 512, 640) float32
├── Z8hl (1, 5, 512, 640) float32
├── Z9hl (1, 5, 512, 640) float32
├── hrrr_x (640,) float64
├── hrrr_y (512,) float64
├── lead_time (5,) timedelta64
├── mslp (1, 5, 512, 640) float32
├── p10hl (1, 5, 512, 640) float32
├── p11hl (1, 5, 512, 640) float32
├── p13hl (1, 5, 512, 640) float32
├── p15hl (1, 5, 512, 640) float32
├── p1hl (1, 5, 512, 640) float32
├── p20hl (1, 5, 512, 640) float32
├── p2hl (1, 5, 512, 640) float32
├── p3hl (1, 5, 512, 640) float32
├── p4hl (1, 5, 512, 640) float32
├── p5hl (1, 5, 512, 640) float32
├── p6hl (1, 5, 512, 640) float32
├── p7hl (1, 5, 512, 640) float32
├── p8hl (1, 5, 512, 640) float32
├── p9hl (1, 5, 512, 640) float32
├── q10hl (1, 5, 512, 640) float32
├── q11hl (1, 5, 512, 640) float32
├── q13hl (1, 5, 512, 640) float32
├── q15hl (1, 5, 512, 640) float32
├── q1hl (1, 5, 512, 640) float32
├── q20hl (1, 5, 512, 640) float32
├── q25hl (1, 5, 512, 640) float32
├── q2hl (1, 5, 512, 640) float32
├── q30hl (1, 5, 512, 640) float32
├── q3hl (1, 5, 512, 640) float32
├── q4hl (1, 5, 512, 640) float32
├── q5hl (1, 5, 512, 640) float32
├── q6hl (1, 5, 512, 640) float32
├── q7hl (1, 5, 512, 640) float32
├── q8hl (1, 5, 512, 640) float32
├── q9hl (1, 5, 512, 640) float32
├── refc (1, 5, 512, 640) float32
├── t10hl (1, 5, 512, 640) float32
├── t11hl (1, 5, 512, 640) float32
├── t13hl (1, 5, 512, 640) float32
├── t15hl (1, 5, 512, 640) float32
├── t1hl (1, 5, 512, 640) float32
├── t20hl (1, 5, 512, 640) float32
├── t25hl (1, 5, 512, 640) float32
├── t2hl (1, 5, 512, 640) float32
├── t2m (1, 5, 512, 640) float32
├── t30hl (1, 5, 512, 640) float32
├── t3hl (1, 5, 512, 640) float32
├── t4hl (1, 5, 512, 640) float32
├── t5hl (1, 5, 512, 640) float32
├── t6hl (1, 5, 512, 640) float32
├── t7hl (1, 5, 512, 640) float32
├── t8hl (1, 5, 512, 640) float32
├── t9hl (1, 5, 512, 640) float32
├── time (1,) datetime64
├── u10hl (1, 5, 512, 640) float32
├── u10m (1, 5, 512, 640) float32
├── u11hl (1, 5, 512, 640) float32
├── u13hl (1, 5, 512, 640) float32
├── u15hl (1, 5, 512, 640) float32
├── u1hl (1, 5, 512, 640) float32
├── u20hl (1, 5, 512, 640) float32
├── u25hl (1, 5, 512, 640) float32
├── u2hl (1, 5, 512, 640) float32
├── u30hl (1, 5, 512, 640) float32
├── u3hl (1, 5, 512, 640) float32
├── u4hl (1, 5, 512, 640) float32
├── u5hl (1, 5, 512, 640) float32
├── u6hl (1, 5, 512, 640) float32
├── u7hl (1, 5, 512, 640) float32
├── u8hl (1, 5, 512, 640) float32
├── u9hl (1, 5, 512, 640) float32
├── v10hl (1, 5, 512, 640) float32
├── v10m (1, 5, 512, 640) float32
├── v11hl (1, 5, 512, 640) float32
├── v13hl (1, 5, 512, 640) float32
├── v15hl (1, 5, 512, 640) float32
├── v1hl (1, 5, 512, 640) float32
├── v20hl (1, 5, 512, 640) float32
├── v25hl (1, 5, 512, 640) float32
├── v2hl (1, 5, 512, 640) float32
├── v30hl (1, 5, 512, 640) float32
├── v3hl (1, 5, 512, 640) float32
├── v4hl (1, 5, 512, 640) float32
├── v5hl (1, 5, 512, 640) float32
├── v6hl (1, 5, 512, 640) float32
├── v7hl (1, 5, 512, 640) float32
├── v8hl (1, 5, 512, 640) float32
└── v9hl (1, 5, 512, 640) float32
Post Processing#
The last step is to post process our results. Cartopy is a great library for plotting fields on projections of a sphere. Here we will just plot the temperature at 2 meters (t2m) 4 hours into the forecast.
Notice that the Zarr IO function has additional APIs to interact with the stored data.
import cartopy
import cartopy.crs as ccrs
import matplotlib.pyplot as plt
forecast = f"{date}"
variable = "t2m"
step = 4 # lead time = 1 hr
plt.close("all")
# Create a correct Lambert Conformal projection
projection = ccrs.LambertConformal(
central_longitude=262.5,
central_latitude=38.5,
standard_parallels=(38.5, 38.5),
globe=ccrs.Globe(semimajor_axis=6371229, semiminor_axis=6371229),
)
# Create a figure and axes with the specified projection
fig, ax = plt.subplots(subplot_kw={"projection": projection}, figsize=(10, 6))
# Plot the field using pcolormesh
im = ax.pcolormesh(
model.lon,
model.lat,
io[variable][0, step],
transform=ccrs.PlateCarree(),
cmap="Spectral_r",
)
# Set state lines
ax.add_feature(
cartopy.feature.STATES.with_scale("50m"), linewidth=0.5, edgecolor="black", zorder=2
)
# Set title
ax.set_title(f"{forecast} - Lead time: {step}hrs")
# Add coastlines and gridlines
ax.coastlines()
ax.gridlines()
plt.savefig(f"outputs/09_{date}_t2m_prediction.jpg")
Total running time of the script: (2 minutes 27.529 seconds)