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
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()
.
def deterministic(
time: list[str] | list[datetime] | list[np.datetime64],
nsteps: int,
prognostic: PrognosticModel,
data: DataSource,
io: IOBackend,
output_coords: CoordSystem = OrderedDict({}),
device: torch.device | None = None,
) -> IOBackend:
"""Built in deterministic workflow.
This workflow creates a determinstic inference pipeline to produce a forecast
prediction using a prognostic model.
Parameters
----------
time : list[str] | list[datetime] | list[np.datetime64]
List of string, datetimes or np.datetime64
nsteps : int
Number of forecast steps
prognostic : PrognosticModel
Prognostic model
data : DataSource
Data source
io : IOBackend
IO object
output_coords: CoordSystem, optional
IO output coordinate system override, by default OrderedDict({})
device : torch.device, optional
Device to run inference on, by default None
Returns
-------
IOBackend
Output IO object
"""
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())
2025-06-16 19:11:31.028 | INFO | earth2studio.run:deterministic:75 - Running simple workflow!
2025-06-16 19:11:31.029 | INFO | earth2studio.run:deterministic:82 - Inference device: cuda
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:31.925 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 133487686-1039560
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.241 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 198761661-2452412
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.281 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 119091744-1068019
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.319 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 105523057-1551698
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.359 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 204280288-906050
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.396 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 366811710-621374
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.431 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 367433084-1247300
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.470 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 13162933-2214901
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.509 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 90842138-1556864
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.549 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 75150938-1107098
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.587 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 76258036-1539245
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.626 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 58340458-2424370
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.666 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 125969662-2207415
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.706 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 19608053-1129808
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.745 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 96746139-2234028
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.786 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 50243470-1061130
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.825 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 89739153-1102985
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.863 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 61868731-1508768
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.903 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 93400931-1032802
Fetching HRRR data: 0%| | 0/99 [00:00<?, ?it/s]
2025-06-16 19:11:32.941 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 140141066-2177271
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:32.981 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 315999554-819908
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.018 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfsfcf00.grib2 25651966-621415
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.055 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 44179015-2400243
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.095 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 6323023-1154066
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.133 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfsfcf00.grib2 36351585-1239980
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.172 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 260702357-842709
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.210 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 258501997-2200360
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.250 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 10142493-1083584
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.288 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 147549410-1009262
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.326 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 67621540-2223847
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.367 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 313536259-696937
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.406 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 9034031-1108462
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.449 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 30662333-2383547
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.489 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 26520279-2215615
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.530 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 194113827-2297871
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.571 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 108077792-1033659
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.609 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 17249831-2358222
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.649 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 22254285-1118061
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.687 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 148558672-1407218
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.726 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 39892416-2217559
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.766 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 78811585-1032403
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.803 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 60764828-1103903
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.842 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 35654313-1105779
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.880 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 134527246-1467067
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.919 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 369524131-855289
Fetching HRRR data: 0%| | 0/99 [00:01<?, ?it/s]
2025-06-16 19:11:33.956 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 314233196-1766358
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:33.996 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfsfcf00.grib2 43576793-2381615
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.023 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 104439522-1083535
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.061 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 178329879-948556
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.100 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfsfcf00.grib2 45958408-2381615
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.127 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 72700071-2450867
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.167 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 53360393-2219125
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.207 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 122693020-1025601
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.245 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 63377499-1030857
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.283 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 77797281-1014304
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.321 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 172607484-2490906
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.361 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 23372346-1089012
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.399 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 33045880-1118945
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.438 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 175098390-976368
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.475 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 87264702-2474451
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.515 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 365334983-1476727
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.555 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 92399002-1001929
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.593 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 150922949-986574
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.630 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 135994313-983842
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.668 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 254409132-2225404
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.708 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 311665652-1870607
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.746 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 149965890-957059
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.783 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 116590483-2501261
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.824 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 64408356-1042688
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.862 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 49171922-1071548
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.900 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 0-2215600
Fetching HRRR data: 0%| | 0/99 [00:02<?, ?it/s]
2025-06-16 19:11:34.940 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 136978155-1014872
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:34.978 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 177404188-925691
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.015 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 111400047-2230071
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.055 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 107074755-1003037
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.093 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 368680384-843747
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.130 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 261545066-1517147
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.169 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 82137456-2229088
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.209 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 121697356-995664
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.246 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 263062213-827592
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.283 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 316819462-820671
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.320 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 130977837-2509849
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.360 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 47687803-1484119
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.399 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 20737861-1516424
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.438 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 145040127-2509283
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.479 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 36760092-1079102
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.516 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 201214073-956581
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.554 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 202170654-1217264
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.592 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 34164825-1489488
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.635 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfsfcf00.grib2 0-312610
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.669 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 263889805-827352
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.706 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 7477089-1556942
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.745 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 46579258-1108545
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.784 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 120159763-1537593
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.823 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 101947962-2491560
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.865 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 176074758-1329430
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.905 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 167654072-2370595
Fetching HRRR data: 0%| | 0/99 [00:03<?, ?it/s]
2025-06-16 19:11:35.946 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 203387918-892370
Fetching HRRR data: 0%| | 0/99 [00:04<?, ?it/s]
2025-06-16 19:11:35.984 | DEBUG | earth2studio.data.hrrr:fetch_array:472 - Fetching HRRR grib file: noaa-hrrr-bdp-pds/hrrr.20250615/conus/hrrr.t00z.wrfnatf00.grib2 4051109-2271914
Fetching HRRR data: 0%| | 0/99 [00:04<?, ?it/s]
Fetching HRRR data: 1%| | 1/99 [00:04<06:41, 4.10s/it]
Fetching HRRR data: 100%|██████████| 99/99 [00:04<00:00, 24.14it/s]
2025-06-16 19:11:36.530 | SUCCESS | earth2studio.run:deterministic:106 - Fetched data from HRRR
2025-06-16 19:11:36.531 | WARNING | earth2studio.io.zarr:add_array:192 - Datetime64 not supported in zarr 3.0, converting to int64 nanoseconds since epoch
2025-06-16 19:11:36.535 | WARNING | earth2studio.io.zarr:add_array:198 - Timedelta64 not supported in zarr 3.0, converting to int64 nanoseconds since epoch
2025-06-16 19:11:36.667 | INFO | earth2studio.run:deterministic:136 - Inference starting!
Running inference: 0%| | 0/5 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2025-06-16 19:11:41.163 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 413369786-838936
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.209 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 0-1001639
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.233 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 263748054-822315
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.256 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 215526320-606159
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.280 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 271112806-954880
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.303 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 432887781-1233676
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.327 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 216132479-617075
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.350 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 407138380-965850
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.374 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 266574515-1328948
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.397 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 401126850-865495
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.420 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 349730587-955521
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.444 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 426057136-957856
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.467 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 209430142-741513
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.490 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 346035276-1273699
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.514 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 212239768-1085011
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.537 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 421462114-508218
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.559 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 270165322-947484
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.583 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 343821867-852313
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.606 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 403008619-1256695
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.630 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 406162384-975996
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.653 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 210171655-760433
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:04<00:17, 4.49s/it]
2025-06-16 19:11:41.676 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 411289139-1002523
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:17, 4.49s/it]
2025-06-16 19:11:41.700 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 264570369-742361
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:17, 4.49s/it]
2025-06-16 19:11:41.723 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 350686108-971978
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:17, 4.49s/it]
2025-06-16 19:11:41.747 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 425082776-974360
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:17, 4.49s/it]
2025-06-16 19:11:41.769 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f000 342911162-910705
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 20%|██ | 1/5 [00:05<00:17, 4.49s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:15, 1.59it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:00<00:00, 41.28it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2025-06-16 19:11:58.126 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 268207265-982171
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.168 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 264622848-1327625
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.192 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 214533720-605847
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.215 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 211526780-1084089
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.238 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 411324018-839220
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.261 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 0-1001057
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.283 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 340815289-910290
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.306 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 269189436-954130
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.329 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 341725579-856040
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.352 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 400941642-1256260
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.375 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 420013062-507861
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.398 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 409243470-1002423
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.421 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 405075244-965577
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.444 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 347637127-957610
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.467 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 215139567-617089
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.490 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 208712335-744004
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.512 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 399060327-864945
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.536 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 261794491-821393
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.559 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 348594737-971601
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.582 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 262615884-741966
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.605 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 209456339-762664
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.627 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 439804056-1233600
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.650 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 425610704-957370
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:21<00:35, 11.83s/it]
2025-06-16 19:11:58.674 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 404096424-978820
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:22<00:35, 11.83s/it]
2025-06-16 19:11:58.697 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 343939770-1273478
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:22<00:35, 11.83s/it]
2025-06-16 19:11:58.720 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f001 424633349-977355
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 40%|████ | 2/5 [00:22<00:35, 11.83s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:15, 1.62it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:00<00:00, 42.11it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2025-06-16 19:12:15.875 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 349561645-957403
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:15.922 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 266909365-1328140
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:15.948 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 216025246-606053
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:15.971 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 212739459-1084824
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:15.995 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 406127078-974152
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.018 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 426687454-973046
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.041 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 342736699-910340
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.064 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 271439008-958168
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.087 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 343647039-856604
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.110 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 402970255-1256257
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.132 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 413369109-838949
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.154 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 411286370-1001894
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.177 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 407101230-964565
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.200 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 270492897-946111
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.223 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 216631299-617342
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.246 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 209936352-740446
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.269 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 401085869-864856
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.291 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 264080107-821178
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.314 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 350519048-970843
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.336 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 264901285-741102
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.359 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 210676798-759761
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.381 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 427660500-956687
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.404 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 0-1000310
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.427 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 442037564-1233248
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.450 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 345864723-1275019
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
2025-06-16 19:12:16.474 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f002 422061780-508449
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 60%|██████ | 3/5 [00:39<00:29, 14.53s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:15, 1.61it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:00<00:00, 41.84it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
2025-06-16 19:12:34.227 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 271339058-945227
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.268 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 267753722-1331513
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.295 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 217116468-605944
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.318 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 214119526-1077233
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.341 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 443304434-1232941
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.364 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 423149708-508548
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.385 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 343900113-910696
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.408 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 272284285-957931
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.430 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 344810809-854463
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.453 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 404122744-1256267
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.476 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 407280735-974033
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.499 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 412407956-1002461
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.522 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 408254768-964944
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.545 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 350724995-953924
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.568 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 217722412-617236
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.591 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 211316618-740199
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.613 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 402235066-864686
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.636 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 264923703-821338
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.659 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 351678919-970276
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:57<00:16, 16.04s/it]
2025-06-16 19:12:34.681 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 265745041-740476
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:58<00:16, 16.04s/it]
2025-06-16 19:12:34.704 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 212056817-762619
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:58<00:16, 16.04s/it]
2025-06-16 19:12:34.726 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 0-1000643
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:58<00:16, 16.04s/it]
2025-06-16 19:12:34.749 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 427765657-973678
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:58<00:16, 16.04s/it]
2025-06-16 19:12:34.771 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 428739335-957347
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:58<00:16, 16.04s/it]
2025-06-16 19:12:34.794 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 347030472-1275742
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:58<00:16, 16.04s/it]
2025-06-16 19:12:34.817 | DEBUG | earth2studio.data.gfs:fetch_array:380 - Fetching GFS grib file: noaa-gfs-bdp-pds/gfs.20250615/00/atmos/gfs.t00z.pgrb2.0p25.f003 414496742-839384
Fetching GFS data: 0%| | 0/26 [00:00<?, ?it/s]
Running inference: 80%|████████ | 4/5 [00:58<00:16, 16.04s/it]
Fetching GFS data: 4%|▍ | 1/26 [00:00<00:15, 1.63it/s]
Fetching GFS data: 100%|██████████| 26/26 [00:00<00:00, 42.44it/s]
Running inference: 100%|██████████| 5/5 [01:16<00:00, 17.08s/it]
Running inference: 100%|██████████| 5/5 [01:16<00:00, 15.30s/it]
2025-06-16 19:12:53.152 | SUCCESS | earth2studio.run:deterministic:146 - 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,) int64
├── hrrr_y (512,) int64
├── lead_time (5,) int64
├── 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,) int64
├── 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: (1 minutes 27.509 seconds)