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

https://arxiv.org/abs/2408.10958

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, but a non-forecast data source such as ARCO could also be used with appropriate time stamps.

from datetime import datetime
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 GFS_FX, HRRR
from earth2studio.io import ZarrBackend
from earth2studio.models.px import StormCast

# Load the default model package which downloads the check point from NGC
package = StormCast.load_default_package()
model = StormCast.load_model(package)

# Create the data source
data = HRRR()

# Create and set the conditioning data source
conditioning_data_source = GFS_FX()
model.conditioning_data_source = conditioning_data_source

# 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()
date = today.isoformat().split("T")[0]
io = run.deterministic([date], nsteps, model, data, io)

print(io.root.tree())

2025-01-23 04:45:34.761 | INFO     | earth2studio.run:deterministic:75 - Running simple workflow!
2025-01-23 04:45:34.762 | INFO     | earth2studio.run:deterministic:82 - Inference device: cuda

Fetching HRRR data:   0%|          | 0/99 [00:00<?, ?it/s]

2025-01-23 04:45:34.903 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u10m at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   0%|          | 0/99 [00:00<?, ?it/s]
Fetching HRRR data:   1%|          | 1/99 [00:00<01:16,  1.29it/s]

2025-01-23 04:45:35.680 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v10m at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   1%|          | 1/99 [00:00<01:16,  1.29it/s]
Fetching HRRR data:   2%|▏         | 2/99 [00:00<00:36,  2.64it/s]

2025-01-23 04:45:35.781 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t2m at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   2%|▏         | 2/99 [00:00<00:36,  2.64it/s]
Fetching HRRR data:   3%|▎         | 3/99 [00:00<00:24,  3.88it/s]

2025-01-23 04:45:35.895 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: mslp at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   3%|▎         | 3/99 [00:00<00:24,  3.88it/s]
Fetching HRRR data:   4%|▍         | 4/99 [00:01<00:18,  5.01it/s]

2025-01-23 04:45:36.005 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u1hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   4%|▍         | 4/99 [00:01<00:18,  5.01it/s]
Fetching HRRR data:   5%|▌         | 5/99 [00:01<00:15,  5.90it/s]

2025-01-23 04:45:36.121 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u2hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   5%|▌         | 5/99 [00:01<00:15,  5.90it/s]
Fetching HRRR data:   6%|▌         | 6/99 [00:01<00:14,  6.62it/s]

2025-01-23 04:45:36.236 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u3hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   6%|▌         | 6/99 [00:01<00:14,  6.62it/s]
Fetching HRRR data:   7%|▋         | 7/99 [00:01<00:12,  7.16it/s]

2025-01-23 04:45:36.352 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u4hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   7%|▋         | 7/99 [00:01<00:12,  7.16it/s]
Fetching HRRR data:   8%|▊         | 8/99 [00:01<00:12,  7.57it/s]

2025-01-23 04:45:36.468 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u5hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   8%|▊         | 8/99 [00:01<00:12,  7.57it/s]
Fetching HRRR data:   9%|▉         | 9/99 [00:01<00:11,  7.88it/s]

2025-01-23 04:45:36.584 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u6hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:   9%|▉         | 9/99 [00:01<00:11,  7.88it/s]
Fetching HRRR data:  10%|█         | 10/99 [00:01<00:10,  8.13it/s]

2025-01-23 04:45:36.698 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u7hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  10%|█         | 10/99 [00:01<00:10,  8.13it/s]
Fetching HRRR data:  11%|█         | 11/99 [00:01<00:10,  8.27it/s]

2025-01-23 04:45:36.814 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u8hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  11%|█         | 11/99 [00:01<00:10,  8.27it/s]
Fetching HRRR data:  12%|█▏        | 12/99 [00:02<00:10,  8.42it/s]

2025-01-23 04:45:36.928 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u9hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  12%|█▏        | 12/99 [00:02<00:10,  8.42it/s]
Fetching HRRR data:  13%|█▎        | 13/99 [00:02<00:10,  8.51it/s]

2025-01-23 04:45:37.042 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u10hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  13%|█▎        | 13/99 [00:02<00:10,  8.51it/s]
Fetching HRRR data:  14%|█▍        | 14/99 [00:02<00:09,  8.58it/s]

2025-01-23 04:45:37.157 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u11hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  14%|█▍        | 14/99 [00:02<00:09,  8.58it/s]
Fetching HRRR data:  15%|█▌        | 15/99 [00:02<00:09,  8.59it/s]

2025-01-23 04:45:37.273 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u13hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  15%|█▌        | 15/99 [00:02<00:09,  8.59it/s]
Fetching HRRR data:  16%|█▌        | 16/99 [00:02<00:09,  8.63it/s]

2025-01-23 04:45:37.388 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u15hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  16%|█▌        | 16/99 [00:02<00:09,  8.63it/s]
Fetching HRRR data:  17%|█▋        | 17/99 [00:02<00:09,  8.73it/s]

2025-01-23 04:45:37.499 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u20hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  17%|█▋        | 17/99 [00:02<00:09,  8.73it/s]
Fetching HRRR data:  18%|█▊        | 18/99 [00:02<00:09,  8.68it/s]

2025-01-23 04:45:37.616 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u25hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  18%|█▊        | 18/99 [00:02<00:09,  8.68it/s]
Fetching HRRR data:  19%|█▉        | 19/99 [00:02<00:09,  8.19it/s]

2025-01-23 04:45:37.754 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: u30hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  19%|█▉        | 19/99 [00:02<00:09,  8.19it/s]
Fetching HRRR data:  20%|██        | 20/99 [00:02<00:10,  7.81it/s]

2025-01-23 04:45:37.896 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v1hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  20%|██        | 20/99 [00:02<00:10,  7.81it/s]
Fetching HRRR data:  21%|██        | 21/99 [00:03<00:09,  8.00it/s]

2025-01-23 04:45:38.014 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v2hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  21%|██        | 21/99 [00:03<00:09,  8.00it/s]
Fetching HRRR data:  22%|██▏       | 22/99 [00:03<00:09,  8.17it/s]

2025-01-23 04:45:38.130 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v3hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  22%|██▏       | 22/99 [00:03<00:09,  8.17it/s]
Fetching HRRR data:  23%|██▎       | 23/99 [00:03<00:09,  8.34it/s]

2025-01-23 04:45:38.244 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v4hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  23%|██▎       | 23/99 [00:03<00:09,  8.34it/s]
Fetching HRRR data:  24%|██▍       | 24/99 [00:03<00:08,  8.42it/s]

2025-01-23 04:45:38.360 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v5hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  24%|██▍       | 24/99 [00:03<00:08,  8.42it/s]
Fetching HRRR data:  25%|██▌       | 25/99 [00:03<00:09,  8.21it/s]

2025-01-23 04:45:38.489 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v6hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  25%|██▌       | 25/99 [00:03<00:09,  8.21it/s]
Fetching HRRR data:  26%|██▋       | 26/99 [00:03<00:08,  8.35it/s]

2025-01-23 04:45:38.604 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v7hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  26%|██▋       | 26/99 [00:03<00:08,  8.35it/s]
Fetching HRRR data:  27%|██▋       | 27/99 [00:03<00:08,  8.21it/s]

2025-01-23 04:45:38.731 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v8hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  27%|██▋       | 27/99 [00:03<00:08,  8.21it/s]
Fetching HRRR data:  28%|██▊       | 28/99 [00:03<00:08,  8.28it/s]

2025-01-23 04:45:38.849 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v9hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  28%|██▊       | 28/99 [00:03<00:08,  8.28it/s]
Fetching HRRR data:  29%|██▉       | 29/99 [00:04<00:08,  8.39it/s]

2025-01-23 04:45:38.964 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v10hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  29%|██▉       | 29/99 [00:04<00:08,  8.39it/s]
Fetching HRRR data:  30%|███       | 30/99 [00:04<00:08,  8.48it/s]

2025-01-23 04:45:39.080 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v11hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  30%|███       | 30/99 [00:04<00:08,  8.48it/s]
Fetching HRRR data:  31%|███▏      | 31/99 [00:04<00:07,  8.55it/s]

2025-01-23 04:45:39.194 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v13hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  31%|███▏      | 31/99 [00:04<00:07,  8.55it/s]
Fetching HRRR data:  32%|███▏      | 32/99 [00:04<00:07,  8.62it/s]

2025-01-23 04:45:39.308 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v15hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  32%|███▏      | 32/99 [00:04<00:07,  8.62it/s]
Fetching HRRR data:  33%|███▎      | 33/99 [00:04<00:07,  8.65it/s]

2025-01-23 04:45:39.423 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v20hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  33%|███▎      | 33/99 [00:04<00:07,  8.65it/s]
Fetching HRRR data:  34%|███▍      | 34/99 [00:04<00:07,  8.71it/s]

2025-01-23 04:45:39.536 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v25hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  34%|███▍      | 34/99 [00:04<00:07,  8.71it/s]
Fetching HRRR data:  35%|███▌      | 35/99 [00:04<00:07,  8.77it/s]

2025-01-23 04:45:39.648 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: v30hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  35%|███▌      | 35/99 [00:04<00:07,  8.77it/s]
Fetching HRRR data:  36%|███▋      | 36/99 [00:04<00:07,  8.78it/s]

2025-01-23 04:45:39.761 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t1hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  36%|███▋      | 36/99 [00:04<00:07,  8.78it/s]
Fetching HRRR data:  37%|███▋      | 37/99 [00:04<00:07,  8.75it/s]

2025-01-23 04:45:39.877 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t2hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  37%|███▋      | 37/99 [00:04<00:07,  8.75it/s]
Fetching HRRR data:  38%|███▊      | 38/99 [00:05<00:07,  8.49it/s]

2025-01-23 04:45:40.003 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t3hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  38%|███▊      | 38/99 [00:05<00:07,  8.49it/s]
Fetching HRRR data:  39%|███▉      | 39/99 [00:05<00:07,  8.54it/s]

2025-01-23 04:45:40.118 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t4hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  39%|███▉      | 39/99 [00:05<00:07,  8.54it/s]
Fetching HRRR data:  40%|████      | 40/99 [00:05<00:07,  8.37it/s]

2025-01-23 04:45:40.243 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t5hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  40%|████      | 40/99 [00:05<00:07,  8.37it/s]
Fetching HRRR data:  41%|████▏     | 41/99 [00:05<00:06,  8.49it/s]

2025-01-23 04:45:40.357 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t6hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  41%|████▏     | 41/99 [00:05<00:06,  8.49it/s]
Fetching HRRR data:  42%|████▏     | 42/99 [00:05<00:06,  8.57it/s]

2025-01-23 04:45:40.471 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t7hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  42%|████▏     | 42/99 [00:05<00:06,  8.57it/s]
Fetching HRRR data:  43%|████▎     | 43/99 [00:05<00:06,  8.40it/s]

2025-01-23 04:45:40.596 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t8hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  43%|████▎     | 43/99 [00:05<00:06,  8.40it/s]
Fetching HRRR data:  44%|████▍     | 44/99 [00:05<00:06,  8.50it/s]

2025-01-23 04:45:40.710 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t9hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  44%|████▍     | 44/99 [00:05<00:06,  8.50it/s]
Fetching HRRR data:  45%|████▌     | 45/99 [00:05<00:06,  8.57it/s]

2025-01-23 04:45:40.825 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t10hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  45%|████▌     | 45/99 [00:05<00:06,  8.57it/s]
Fetching HRRR data:  46%|████▋     | 46/99 [00:06<00:06,  8.61it/s]

2025-01-23 04:45:40.939 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t11hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  46%|████▋     | 46/99 [00:06<00:06,  8.61it/s]
Fetching HRRR data:  47%|████▋     | 47/99 [00:06<00:06,  8.64it/s]

2025-01-23 04:45:41.054 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t13hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  47%|████▋     | 47/99 [00:06<00:06,  8.64it/s]
Fetching HRRR data:  48%|████▊     | 48/99 [00:06<00:05,  8.66it/s]

2025-01-23 04:45:41.169 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t15hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  48%|████▊     | 48/99 [00:06<00:05,  8.66it/s]
Fetching HRRR data:  49%|████▉     | 49/99 [00:06<00:05,  8.46it/s]

2025-01-23 04:45:41.294 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t20hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  49%|████▉     | 49/99 [00:06<00:05,  8.46it/s]
Fetching HRRR data:  51%|█████     | 50/99 [00:06<00:05,  8.33it/s]

2025-01-23 04:45:41.418 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t25hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  51%|█████     | 50/99 [00:06<00:05,  8.33it/s]
Fetching HRRR data:  52%|█████▏    | 51/99 [00:06<00:05,  8.46it/s]

2025-01-23 04:45:41.532 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: t30hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  52%|█████▏    | 51/99 [00:06<00:05,  8.46it/s]
Fetching HRRR data:  53%|█████▎    | 52/99 [00:06<00:05,  8.58it/s]

2025-01-23 04:45:41.644 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q1hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  53%|█████▎    | 52/99 [00:06<00:05,  8.58it/s]
Fetching HRRR data:  54%|█████▎    | 53/99 [00:06<00:05,  8.58it/s]

2025-01-23 04:45:41.761 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q2hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  54%|█████▎    | 53/99 [00:06<00:05,  8.58it/s]
Fetching HRRR data:  55%|█████▍    | 54/99 [00:06<00:05,  8.61it/s]

2025-01-23 04:45:41.876 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q3hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  55%|█████▍    | 54/99 [00:06<00:05,  8.61it/s]
Fetching HRRR data:  56%|█████▌    | 55/99 [00:07<00:05,  8.60it/s]

2025-01-23 04:45:41.993 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q4hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  56%|█████▌    | 55/99 [00:07<00:05,  8.60it/s]
Fetching HRRR data:  57%|█████▋    | 56/99 [00:07<00:04,  8.63it/s]

2025-01-23 04:45:42.108 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q5hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  57%|█████▋    | 56/99 [00:07<00:04,  8.63it/s]
Fetching HRRR data:  58%|█████▊    | 57/99 [00:07<00:05,  8.38it/s]

2025-01-23 04:45:42.235 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q6hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  58%|█████▊    | 57/99 [00:07<00:05,  8.38it/s]
Fetching HRRR data:  59%|█████▊    | 58/99 [00:07<00:04,  8.34it/s]

2025-01-23 04:45:42.356 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q7hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  59%|█████▊    | 58/99 [00:07<00:04,  8.34it/s]
Fetching HRRR data:  60%|█████▉    | 59/99 [00:07<00:04,  8.19it/s]

2025-01-23 04:45:42.483 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q8hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  60%|█████▉    | 59/99 [00:07<00:04,  8.19it/s]
Fetching HRRR data:  61%|██████    | 60/99 [00:07<00:04,  8.34it/s]

2025-01-23 04:45:42.599 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q9hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  61%|██████    | 60/99 [00:07<00:04,  8.34it/s]
Fetching HRRR data:  62%|██████▏   | 61/99 [00:07<00:04,  8.45it/s]

2025-01-23 04:45:42.713 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q10hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  62%|██████▏   | 61/99 [00:07<00:04,  8.45it/s]
Fetching HRRR data:  63%|██████▎   | 62/99 [00:07<00:04,  8.27it/s]

2025-01-23 04:45:42.840 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q11hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  63%|██████▎   | 62/99 [00:07<00:04,  8.27it/s]
Fetching HRRR data:  64%|██████▎   | 63/99 [00:08<00:04,  8.38it/s]

2025-01-23 04:45:42.956 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q13hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  64%|██████▎   | 63/99 [00:08<00:04,  8.38it/s]
Fetching HRRR data:  65%|██████▍   | 64/99 [00:08<00:04,  8.40it/s]

2025-01-23 04:45:43.074 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q15hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  65%|██████▍   | 64/99 [00:08<00:04,  8.40it/s]
Fetching HRRR data:  66%|██████▌   | 65/99 [00:08<00:04,  8.25it/s]

2025-01-23 04:45:43.200 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q20hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  66%|██████▌   | 65/99 [00:08<00:04,  8.25it/s]
Fetching HRRR data:  67%|██████▋   | 66/99 [00:08<00:03,  8.39it/s]

2025-01-23 04:45:43.315 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q25hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  67%|██████▋   | 66/99 [00:08<00:03,  8.39it/s]
Fetching HRRR data:  68%|██████▊   | 67/99 [00:08<00:03,  8.49it/s]

2025-01-23 04:45:43.429 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: q30hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  68%|██████▊   | 67/99 [00:08<00:03,  8.49it/s]
Fetching HRRR data:  69%|██████▊   | 68/99 [00:08<00:03,  8.51it/s]

2025-01-23 04:45:43.546 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z1hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  69%|██████▊   | 68/99 [00:08<00:03,  8.51it/s]
Fetching HRRR data:  70%|██████▉   | 69/99 [00:08<00:03,  8.53it/s]

2025-01-23 04:45:43.663 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z2hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  70%|██████▉   | 69/99 [00:08<00:03,  8.53it/s]
Fetching HRRR data:  71%|███████   | 70/99 [00:08<00:03,  8.52it/s]

2025-01-23 04:45:43.780 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z3hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  71%|███████   | 70/99 [00:08<00:03,  8.52it/s]
Fetching HRRR data:  72%|███████▏  | 71/99 [00:08<00:03,  8.53it/s]

2025-01-23 04:45:43.897 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z4hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  72%|███████▏  | 71/99 [00:08<00:03,  8.53it/s]
Fetching HRRR data:  73%|███████▎  | 72/99 [00:09<00:03,  8.51it/s]

2025-01-23 04:45:44.015 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z5hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  73%|███████▎  | 72/99 [00:09<00:03,  8.51it/s]
Fetching HRRR data:  74%|███████▎  | 73/99 [00:09<00:03,  8.30it/s]

2025-01-23 04:45:44.143 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z6hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  74%|███████▎  | 73/99 [00:09<00:03,  8.30it/s]
Fetching HRRR data:  75%|███████▍  | 74/99 [00:09<00:02,  8.36it/s]

2025-01-23 04:45:44.261 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z7hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  75%|███████▍  | 74/99 [00:09<00:02,  8.36it/s]
Fetching HRRR data:  76%|███████▌  | 75/99 [00:09<00:02,  8.37it/s]

2025-01-23 04:45:44.379 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z8hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  76%|███████▌  | 75/99 [00:09<00:02,  8.37it/s]
Fetching HRRR data:  77%|███████▋  | 76/99 [00:09<00:02,  8.36it/s]

2025-01-23 04:45:44.500 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z9hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  77%|███████▋  | 76/99 [00:09<00:02,  8.36it/s]
Fetching HRRR data:  78%|███████▊  | 77/99 [00:09<00:02,  8.38it/s]

2025-01-23 04:45:44.618 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z10hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  78%|███████▊  | 77/99 [00:09<00:02,  8.38it/s]
Fetching HRRR data:  79%|███████▉  | 78/99 [00:09<00:02,  8.42it/s]

2025-01-23 04:45:44.736 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z11hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  79%|███████▉  | 78/99 [00:09<00:02,  8.42it/s]
Fetching HRRR data:  80%|███████▉  | 79/99 [00:09<00:02,  8.44it/s]

2025-01-23 04:45:44.854 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z13hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  80%|███████▉  | 79/99 [00:09<00:02,  8.44it/s]
Fetching HRRR data:  81%|████████  | 80/99 [00:10<00:02,  8.42it/s]

2025-01-23 04:45:44.973 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z15hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  81%|████████  | 80/99 [00:10<00:02,  8.42it/s]
Fetching HRRR data:  82%|████████▏ | 81/99 [00:10<00:02,  8.45it/s]

2025-01-23 04:45:45.091 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z20hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  82%|████████▏ | 81/99 [00:10<00:02,  8.45it/s]
Fetching HRRR data:  83%|████████▎ | 82/99 [00:10<00:02,  8.44it/s]

2025-01-23 04:45:45.209 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z25hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  83%|████████▎ | 82/99 [00:10<00:02,  8.44it/s]
Fetching HRRR data:  84%|████████▍ | 83/99 [00:10<00:01,  8.49it/s]

2025-01-23 04:45:45.325 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: Z30hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  84%|████████▍ | 83/99 [00:10<00:01,  8.49it/s]
Fetching HRRR data:  85%|████████▍ | 84/99 [00:10<00:01,  8.30it/s]

2025-01-23 04:45:45.452 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p1hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  85%|████████▍ | 84/99 [00:10<00:01,  8.30it/s]
Fetching HRRR data:  86%|████████▌ | 85/99 [00:10<00:01,  8.33it/s]

2025-01-23 04:45:45.571 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p2hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  86%|████████▌ | 85/99 [00:10<00:01,  8.33it/s]
Fetching HRRR data:  87%|████████▋ | 86/99 [00:10<00:01,  8.38it/s]

2025-01-23 04:45:45.689 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p3hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  87%|████████▋ | 86/99 [00:10<00:01,  8.38it/s]
Fetching HRRR data:  88%|████████▊ | 87/99 [00:10<00:01,  8.42it/s]

2025-01-23 04:45:45.806 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p4hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  88%|████████▊ | 87/99 [00:10<00:01,  8.42it/s]
Fetching HRRR data:  89%|████████▉ | 88/99 [00:11<00:01,  8.45it/s]

2025-01-23 04:45:45.924 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p5hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  89%|████████▉ | 88/99 [00:11<00:01,  8.45it/s]
Fetching HRRR data:  90%|████████▉ | 89/99 [00:11<00:01,  8.27it/s]

2025-01-23 04:45:46.051 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p6hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  90%|████████▉ | 89/99 [00:11<00:01,  8.27it/s]
Fetching HRRR data:  91%|█████████ | 90/99 [00:11<00:01,  8.34it/s]

2025-01-23 04:45:46.168 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p7hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  91%|█████████ | 90/99 [00:11<00:01,  8.34it/s]
Fetching HRRR data:  92%|█████████▏| 91/99 [00:11<00:00,  8.40it/s]

2025-01-23 04:45:46.285 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p8hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  92%|█████████▏| 91/99 [00:11<00:00,  8.40it/s]
Fetching HRRR data:  93%|█████████▎| 92/99 [00:11<00:00,  8.46it/s]

2025-01-23 04:45:46.402 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p9hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  93%|█████████▎| 92/99 [00:11<00:00,  8.46it/s]
Fetching HRRR data:  94%|█████████▍| 93/99 [00:11<00:00,  8.47it/s]

2025-01-23 04:45:46.519 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p10hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  94%|█████████▍| 93/99 [00:11<00:00,  8.47it/s]
Fetching HRRR data:  95%|█████████▍| 94/99 [00:11<00:00,  8.24it/s]

2025-01-23 04:45:46.648 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p11hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  95%|█████████▍| 94/99 [00:11<00:00,  8.24it/s]
Fetching HRRR data:  96%|█████████▌| 95/99 [00:11<00:00,  8.12it/s]

2025-01-23 04:45:46.775 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p13hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  96%|█████████▌| 95/99 [00:11<00:00,  8.12it/s]
Fetching HRRR data:  97%|█████████▋| 96/99 [00:12<00:00,  8.01it/s]

2025-01-23 04:45:46.904 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p15hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  97%|█████████▋| 96/99 [00:12<00:00,  8.01it/s]
Fetching HRRR data:  98%|█████████▊| 97/99 [00:12<00:00,  7.97it/s]

2025-01-23 04:45:47.031 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: p20hl at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  98%|█████████▊| 97/99 [00:12<00:00,  7.97it/s]
Fetching HRRR data:  99%|█████████▉| 98/99 [00:12<00:00,  8.12it/s]

2025-01-23 04:45:47.149 | DEBUG    | earth2studio.data.hrrr:async_fetch:130 - Fetching HRRR data for variable: refc at 2025-01-23T00:00:00 lead time 0:00:00

Fetching HRRR data:  99%|█████████▉| 98/99 [00:12<00:00,  8.12it/s]
Fetching HRRR data: 100%|██████████| 99/99 [00:12<00:00,  8.47it/s]
Fetching HRRR data: 100%|██████████| 99/99 [00:12<00:00,  8.01it/s]
2025-01-23 04:46:43.183 | SUCCESS  | earth2studio.run:deterministic:106 - Fetched data from HRRR
2025-01-23 04:46:43.306 | INFO     | earth2studio.run:deterministic:136 - Inference starting!

Running inference:   0%|          | 0/5 [00:00<?, ?it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.468 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:209 - Fetching GFS index file: 2025-01-23 00:00:00 lead 0:00:00

Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]

2025-01-23 04:46:44.471 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u10m at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.498 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v10m at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.524 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t2m at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.551 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: tcwv at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.64it/s]

2025-01-23 04:46:44.578 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: msl at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.64it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.603 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: sp at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.64it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.629 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u1000 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.64it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.655 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u850 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.64it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.18it/s]

2025-01-23 04:46:44.682 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u500 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.18it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.707 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u250 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.18it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.732 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v1000 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.18it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.759 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v850 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.18it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.38it/s]

2025-01-23 04:46:44.785 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v500 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.38it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.811 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v250 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.38it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.836 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z1000 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.38it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.862 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z850 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.38it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.26it/s]

2025-01-23 04:46:44.890 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z500 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.26it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.916 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z250 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.26it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.942 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t1000 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.26it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:44.968 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t850 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.26it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.33it/s]

2025-01-23 04:46:44.994 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t500 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.33it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:45.020 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t250 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.33it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:45.045 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q1000 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.33it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:45.071 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q850 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.33it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.43it/s]

2025-01-23 04:46:45.098 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q500 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.43it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]

2025-01-23 04:46:45.124 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q250 at 2025-01-23 00:00:00_0:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.43it/s]
Running inference:  20%|██        | 1/5 [00:01<00:04,  1.16s/it]
Fetching GFS for 2025-01-23 00:00:00: 100%|██████████| 26/26 [00:00<00:00, 38.31it/s]

Running inference:  40%|████      | 2/5 [00:20<00:36, 12.07s/it]

2025-01-23 04:47:04.172 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:209 - Fetching GFS index file: 2025-01-23 00:00:00 lead 1:00:00

Running inference:  40%|████      | 2/5 [00:20<00:36, 12.07s/it]

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]

2025-01-23 04:47:04.176 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u10m at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  40%|████      | 2/5 [00:20<00:36, 12.07s/it]

2025-01-23 04:47:04.202 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v10m at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  40%|████      | 2/5 [00:20<00:36, 12.07s/it]

2025-01-23 04:47:04.228 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t2m at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  40%|████      | 2/5 [00:20<00:36, 12.07s/it]

2025-01-23 04:47:04.256 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: tcwv at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  40%|████      | 2/5 [00:20<00:36, 12.07s/it]

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.74it/s]

2025-01-23 04:47:04.282 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: msl at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.74it/s]
Running inference:  40%|████      | 2/5 [00:20<00:36, 12.07s/it]

2025-01-23 04:47:04.308 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: sp at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.74it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.334 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u1000 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.74it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.359 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u850 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 37.74it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.31it/s]

2025-01-23 04:47:04.386 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u500 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.31it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.411 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u250 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.31it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.436 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v1000 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.31it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.462 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v850 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.31it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.60it/s]

2025-01-23 04:47:04.488 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v500 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.60it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.513 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v250 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.60it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.538 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z1000 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.60it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.564 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z850 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.60it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.76it/s]

2025-01-23 04:47:04.591 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z500 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.76it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.616 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z250 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.76it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.642 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t1000 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.76it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.668 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t850 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.76it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.71it/s]

2025-01-23 04:47:04.694 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t500 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.71it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.720 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t250 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.71it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.745 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q1000 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.71it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.772 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q850 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.71it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.63it/s]

2025-01-23 04:47:04.798 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q500 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.63it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]

2025-01-23 04:47:04.824 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q250 at 2025-01-23 00:00:00_1:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.63it/s]
Running inference:  40%|████      | 2/5 [00:21<00:36, 12.07s/it]
Fetching GFS for 2025-01-23 00:00:00: 100%|██████████| 26/26 [00:00<00:00, 38.54it/s]

Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.005 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:209 - Fetching GFS index file: 2025-01-23 00:00:00 lead 2:00:00

Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]

2025-01-23 04:47:24.009 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u10m at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.036 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v10m at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.062 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t2m at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.088 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: tcwv at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 38.07it/s]

2025-01-23 04:47:24.114 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: msl at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 38.07it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.140 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: sp at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 38.07it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.166 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u1000 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 38.07it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.192 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u850 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 38.07it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.28it/s]

2025-01-23 04:47:24.219 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u500 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.28it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.244 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u250 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.28it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.270 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v1000 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.28it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

2025-01-23 04:47:24.296 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v850 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 38.28it/s]
Running inference:  60%|██████    | 3/5 [00:40<00:31, 15.61s/it]

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.44it/s]

2025-01-23 04:47:24.322 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v500 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.44it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.347 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v250 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.44it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.373 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z1000 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.44it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.399 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z850 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 38.44it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.49it/s]

2025-01-23 04:47:24.426 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z500 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.49it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.452 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z250 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.49it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.477 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t1000 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.49it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.503 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t850 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 38.49it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.51it/s]

2025-01-23 04:47:24.530 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t500 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.51it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.555 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t250 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.51it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.581 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q1000 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.51it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.607 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q850 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 38.51it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.44it/s]

2025-01-23 04:47:24.634 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q500 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.44it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]

2025-01-23 04:47:24.660 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q250 at 2025-01-23 00:00:00_2:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 38.44it/s]
Running inference:  60%|██████    | 3/5 [00:41<00:31, 15.61s/it]
Fetching GFS for 2025-01-23 00:00:00: 100%|██████████| 26/26 [00:00<00:00, 38.36it/s]

Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:43.967 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:209 - Fetching GFS index file: 2025-01-23 00:00:00 lead 3:00:00

Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]

2025-01-23 04:47:43.971 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u10m at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:43.999 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v10m at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.026 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t2m at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.053 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: tcwv at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 36.92it/s]

2025-01-23 04:47:44.080 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: msl at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 36.92it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.106 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: sp at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 36.92it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.133 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u1000 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 36.92it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.159 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u850 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  15%|█▌        | 4/26 [00:00<00:00, 36.92it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 37.43it/s]

2025-01-23 04:47:44.186 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u500 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 37.43it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.212 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: u250 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 37.43it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.238 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v1000 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 37.43it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.265 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v850 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  31%|███       | 8/26 [00:00<00:00, 37.43it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 37.54it/s]

2025-01-23 04:47:44.292 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v500 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 37.54it/s]
Running inference:  80%|████████  | 4/5 [01:00<00:17, 17.33s/it]

2025-01-23 04:47:44.318 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: v250 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 37.54it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.344 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z1000 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 37.54it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.370 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z850 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 37.54it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 37.71it/s]

2025-01-23 04:47:44.397 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z500 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 37.71it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.424 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: z250 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 37.71it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.450 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t1000 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 37.71it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.477 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t850 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 37.71it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 37.58it/s]

2025-01-23 04:47:44.504 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t500 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 37.58it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.531 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: t250 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 37.58it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.557 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q1000 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 37.58it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.584 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q850 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 37.58it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 37.47it/s]

2025-01-23 04:47:44.612 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q500 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 37.47it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]

2025-01-23 04:47:44.638 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:255 - Fetching GFS grib file for variable: q250 at 2025-01-23 00:00:00_3:00:00

Fetching GFS for 2025-01-23 00:00:00:  92%|█████████▏| 24/26 [00:00<00:00, 37.47it/s]
Running inference:  80%|████████  | 4/5 [01:01<00:17, 17.33s/it]
Fetching GFS for 2025-01-23 00:00:00: 100%|██████████| 26/26 [00:00<00:00, 37.46it/s]

Running inference: 100%|██████████| 5/5 [01:20<00:00, 18.33s/it]
Running inference: 100%|██████████| 5/5 [01:20<00:00, 16.15s/it]
2025-01-23 04:48:04.059 | 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
 ├── ilat (512,) int64
 ├── ilon (640,) int64
 ├── lat (512, 640) float32
 ├── lead_time (5,) timedelta64[h]
 ├── lon (512, 640) float32
 ├── 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[ns]
 ├── 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(
    io["lon"][:],
    io["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 35.466 seconds)

Gallery generated by Sphinx-Gallery