.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "examples/09_stormcast_example.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_examples_09_stormcast_example.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_examples_09_stormcast_example.py:


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

.. GENERATED FROM PYTHON SOURCE LINES 33-38

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:
:py:meth:`earth2studio.run.deterministic`.

.. GENERATED FROM PYTHON SOURCE LINES 40-44

.. literalinclude:: ../../earth2studio/run.py
   :language: python
   :start-after: # sphinx - deterministic start
   :end-before: # sphinx - deterministic end

.. GENERATED FROM PYTHON SOURCE LINES 46-55

Thus, we need the following:

- Prognostic Model: Use the built in StormCast Model :py:class:`earth2studio.models.px.StormCast`.
- Datasource: Pull data from the HRRR data api :py:class:`earth2studio.data.HRRR`.
- IO Backend: Let's save the outputs into a Zarr store :py:class:`earth2studio.io.ZarrBackend`.

StormCast also requires a conditioning data source. We use a forecast data source here,
GFS_FX :py:class:`earth2studio.data.GFS_FX`, but a non-forecast data source such as ARCO
could also be used with appropriate time stamps.

.. GENERATED FROM PYTHON SOURCE LINES 57-90

.. code-block:: Python

    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 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()





.. rst-class:: sphx-glr-script-out

 .. code-block:: none

     ╭─▌▌Herbie─────────────────────────────────────────────╮
     │ INFO: Created a default config file.                 │
     │ You may view/edit Herbie's configuration here:       │
     │          /root/.config/herbie/config.toml            │
     ╰──────────────────────────────────────────────────────╯





.. GENERATED FROM PYTHON SOURCE LINES 91-99

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

.. GENERATED FROM PYTHON SOURCE LINES 101-110

.. code-block:: Python

    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())





.. rst-class:: sphx-glr-script-out

 .. code-block:: none

    2025-03-27 08:01:00.806 | INFO     | earth2studio.run:deterministic:75 - Running simple workflow!
    2025-03-27 08:01:00.807 | INFO     | earth2studio.run:deterministic:82 - Inference device: cuda

    Fetching HRRR data:   0%|          | 0/99 [00:00<?, ?it/s]
    Fetching HRRR data:   1%|          | 1/99 [00:01<02:08,  1.31s/it]
    Fetching HRRR data:   2%|▏         | 2/99 [00:02<01:32,  1.05it/s]
    Fetching HRRR data:   3%|▎         | 3/99 [00:02<01:19,  1.20it/s]
    Fetching HRRR data:   4%|▍         | 4/99 [00:03<01:13,  1.29it/s]
    Fetching HRRR data:   5%|▌         | 5/99 [00:04<01:12,  1.30it/s]
    Fetching HRRR data:   6%|▌         | 6/99 [00:04<01:10,  1.31it/s]
    Fetching HRRR data:   7%|▋         | 7/99 [00:06<01:24,  1.10it/s]
    Fetching HRRR data:   8%|▊         | 8/99 [00:06<01:18,  1.16it/s]
    Fetching HRRR data:   9%|▉         | 9/99 [00:07<01:14,  1.21it/s]
    Fetching HRRR data:  10%|█         | 10/99 [00:08<01:11,  1.24it/s]
    Fetching HRRR data:  11%|█         | 11/99 [00:09<01:09,  1.26it/s]
    Fetching HRRR data:  12%|█▏        | 12/99 [00:09<01:08,  1.27it/s]
    Fetching HRRR data:  13%|█▎        | 13/99 [00:10<01:12,  1.18it/s]
    Fetching HRRR data:  14%|█▍        | 14/99 [00:11<01:14,  1.14it/s]
    Fetching HRRR data:  15%|█▌        | 15/99 [00:12<01:11,  1.18it/s]
    Fetching HRRR data:  16%|█▌        | 16/99 [00:13<01:08,  1.22it/s]
    Fetching HRRR data:  17%|█▋        | 17/99 [00:14<01:06,  1.24it/s]
    Fetching HRRR data:  18%|█▊        | 18/99 [00:14<01:04,  1.26it/s]
    Fetching HRRR data:  19%|█▉        | 19/99 [00:15<01:02,  1.28it/s]
    Fetching HRRR data:  20%|██        | 20/99 [00:16<01:11,  1.11it/s]
    Fetching HRRR data:  21%|██        | 21/99 [00:17<01:07,  1.15it/s]
    Fetching HRRR data:  22%|██▏       | 22/99 [00:18<01:05,  1.18it/s]
    Fetching HRRR data:  23%|██▎       | 23/99 [00:19<01:03,  1.20it/s]
    Fetching HRRR data:  24%|██▍       | 24/99 [00:20<01:00,  1.23it/s]
    Fetching HRRR data:  25%|██▌       | 25/99 [00:20<00:58,  1.27it/s]
    Fetching HRRR data:  26%|██▋       | 26/99 [00:21<01:06,  1.10it/s]
    Fetching HRRR data:  27%|██▋       | 27/99 [00:22<01:01,  1.17it/s]
    Fetching HRRR data:  28%|██▊       | 28/99 [00:23<00:58,  1.22it/s]
    Fetching HRRR data:  29%|██▉       | 29/99 [00:24<00:56,  1.25it/s]
    Fetching HRRR data:  30%|███       | 30/99 [00:24<00:54,  1.27it/s]
    Fetching HRRR data:  31%|███▏      | 31/99 [00:25<00:52,  1.29it/s]
    Fetching HRRR data:  32%|███▏      | 32/99 [00:26<00:56,  1.19it/s]
    Fetching HRRR data:  33%|███▎      | 33/99 [00:27<00:58,  1.13it/s]
    Fetching HRRR data:  34%|███▍      | 34/99 [00:28<00:55,  1.16it/s]
    Fetching HRRR data:  35%|███▌      | 35/99 [00:29<00:53,  1.20it/s]
    Fetching HRRR data:  36%|███▋      | 36/99 [00:29<00:51,  1.22it/s]
    Fetching HRRR data:  37%|███▋      | 37/99 [00:30<00:49,  1.24it/s]
    Fetching HRRR data:  38%|███▊      | 38/99 [00:31<00:52,  1.17it/s]
    Fetching HRRR data:  39%|███▉      | 39/99 [00:32<00:53,  1.12it/s]
    Fetching HRRR data:  40%|████      | 40/99 [00:33<00:50,  1.17it/s]
    Fetching HRRR data:  41%|████▏     | 41/99 [00:34<00:48,  1.21it/s]
    Fetching HRRR data:  42%|████▏     | 42/99 [00:35<00:46,  1.24it/s]
    Fetching HRRR data:  43%|████▎     | 43/99 [00:35<00:44,  1.25it/s]
    Fetching HRRR data:  44%|████▍     | 44/99 [00:36<00:47,  1.15it/s]
    Fetching HRRR data:  45%|████▌     | 45/99 [00:37<00:49,  1.10it/s]
    Fetching HRRR data:  46%|████▋     | 46/99 [00:38<00:45,  1.16it/s]
    Fetching HRRR data:  47%|████▋     | 47/99 [00:39<00:43,  1.19it/s]
    Fetching HRRR data:  48%|████▊     | 48/99 [00:40<00:42,  1.20it/s]
    Fetching HRRR data:  49%|████▉     | 49/99 [00:40<00:40,  1.23it/s]
    Fetching HRRR data:  51%|█████     | 50/99 [00:41<00:42,  1.16it/s]
    Fetching HRRR data:  52%|█████▏    | 51/99 [00:42<00:42,  1.12it/s]
    Fetching HRRR data:  53%|█████▎    | 52/99 [00:43<00:39,  1.18it/s]
    Fetching HRRR data:  54%|█████▎    | 53/99 [00:44<00:38,  1.20it/s]
    Fetching HRRR data:  55%|█████▍    | 54/99 [00:45<00:36,  1.23it/s]
    Fetching HRRR data:  56%|█████▌    | 55/99 [00:45<00:35,  1.26it/s]
    Fetching HRRR data:  57%|█████▋    | 56/99 [00:46<00:34,  1.26it/s]
    Fetching HRRR data:  58%|█████▊    | 57/99 [00:47<00:38,  1.09it/s]
    Fetching HRRR data:  59%|█████▊    | 58/99 [00:48<00:35,  1.14it/s]
    Fetching HRRR data:  60%|█████▉    | 59/99 [00:49<00:33,  1.18it/s]
    Fetching HRRR data:  61%|██████    | 60/99 [00:50<00:31,  1.22it/s]
    Fetching HRRR data:  62%|██████▏   | 61/99 [00:51<00:30,  1.24it/s]
    Fetching HRRR data:  63%|██████▎   | 62/99 [00:51<00:29,  1.26it/s]
    Fetching HRRR data:  64%|██████▎   | 63/99 [00:52<00:33,  1.09it/s]
    Fetching HRRR data:  65%|██████▍   | 64/99 [00:53<00:30,  1.16it/s]
    Fetching HRRR data:  66%|██████▌   | 65/99 [00:54<00:28,  1.19it/s]
    Fetching HRRR data:  67%|██████▋   | 66/99 [00:55<00:26,  1.22it/s]
    Fetching HRRR data:  68%|██████▊   | 67/99 [00:56<00:25,  1.25it/s]
    Fetching HRRR data:  69%|██████▊   | 68/99 [00:56<00:24,  1.27it/s]
    Fetching HRRR data:  70%|██████▉   | 69/99 [00:57<00:25,  1.17it/s]
    Fetching HRRR data:  71%|███████   | 70/99 [00:58<00:26,  1.07it/s]
    Fetching HRRR data:  72%|███████▏  | 71/99 [00:59<00:25,  1.12it/s]
    Fetching HRRR data:  73%|███████▎  | 72/99 [01:00<00:23,  1.15it/s]
    Fetching HRRR data:  74%|███████▎  | 73/99 [01:01<00:21,  1.19it/s]
    Fetching HRRR data:  75%|███████▍  | 74/99 [01:02<00:20,  1.21it/s]
    Fetching HRRR data:  76%|███████▌  | 75/99 [01:03<00:21,  1.14it/s]
    Fetching HRRR data:  77%|███████▋  | 76/99 [01:04<00:21,  1.09it/s]
    Fetching HRRR data:  78%|███████▊  | 77/99 [01:04<00:19,  1.14it/s]
    Fetching HRRR data:  79%|███████▉  | 78/99 [01:05<00:17,  1.17it/s]
    Fetching HRRR data:  80%|███████▉  | 79/99 [01:06<00:16,  1.19it/s]
    Fetching HRRR data:  81%|████████  | 80/99 [01:07<00:15,  1.20it/s]
    Fetching HRRR data:  82%|████████▏ | 81/99 [01:08<00:16,  1.12it/s]
    Fetching HRRR data:  83%|████████▎ | 82/99 [01:09<00:15,  1.07it/s]
    Fetching HRRR data:  84%|████████▍ | 83/99 [01:10<00:14,  1.13it/s]
    Fetching HRRR data:  85%|████████▍ | 84/99 [01:10<00:12,  1.16it/s]
    Fetching HRRR data:  86%|████████▌ | 85/99 [01:11<00:11,  1.18it/s]
    Fetching HRRR data:  87%|████████▋ | 86/99 [01:12<00:10,  1.19it/s]
    Fetching HRRR data:  88%|████████▊ | 87/99 [01:13<00:10,  1.12it/s]
    Fetching HRRR data:  89%|████████▉ | 88/99 [01:14<00:10,  1.08it/s]
    Fetching HRRR data:  90%|████████▉ | 89/99 [01:15<00:08,  1.12it/s]
    Fetching HRRR data:  91%|█████████ | 90/99 [01:16<00:07,  1.16it/s]
    Fetching HRRR data:  92%|█████████▏| 91/99 [01:17<00:06,  1.19it/s]
    Fetching HRRR data:  93%|█████████▎| 92/99 [01:17<00:05,  1.21it/s]
    Fetching HRRR data:  94%|█████████▍| 93/99 [01:18<00:05,  1.14it/s]
    Fetching HRRR data:  95%|█████████▍| 94/99 [01:19<00:04,  1.08it/s]
    Fetching HRRR data:  96%|█████████▌| 95/99 [01:20<00:03,  1.13it/s]
    Fetching HRRR data:  97%|█████████▋| 96/99 [01:21<00:02,  1.16it/s]
    Fetching HRRR data:  98%|█████████▊| 97/99 [01:22<00:01,  1.19it/s]
    Fetching HRRR data:  99%|█████████▉| 98/99 [01:23<00:00,  1.22it/s]
    Fetching HRRR data: 100%|██████████| 99/99 [01:23<00:00,  1.19it/s]
    Fetching HRRR data: 100%|██████████| 99/99 [01:23<00:00,  1.18it/s]
    2025-03-27 08:03:10.624 | SUCCESS  | earth2studio.run:deterministic:106 - Fetched data from HRRR
    2025-03-27 08:03:10.722 | INFO     | earth2studio.run:deterministic:136 - Inference starting!

    Running inference:   0%|          | 0/5 [00:00<?, ?it/s]
    Running inference:  20%|██        | 1/5 [00:00<00:03,  1.32it/s]
                                                                
    2025-03-27 08:03:11.495 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:214 - Fetching GFS index file: 2025-03-26 00:00:00 lead 0:00:00

    Running inference:  20%|██        | 1/5 [00:00<00:03,  1.32it/s]

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

                                                                            
    
                                                                
    2025-03-27 08:03:11.508 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u10m at 2025-03-26 00:00:00_0:00:00


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

                                                                            
    
                                                                
    2025-03-27 08:03:11.561 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v10m at 2025-03-26 00:00:00_0:00:00


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

    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 19.49it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.611 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t2m at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 19.49it/s]
    Running inference:  20%|██        | 1/5 [00:00<00:03,  1.32it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.654 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: tcwv at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 19.49it/s]
    Running inference:  20%|██        | 1/5 [00:00<00:03,  1.32it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.702 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: msl at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 19.49it/s]
    Running inference:  20%|██        | 1/5 [00:00<00:03,  1.32it/s]

    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.21it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.757 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: sp at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.21it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.796 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u1000 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.21it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.850 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u850 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.21it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

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

                                                                                    
    
                                                                
    2025-03-27 08:03:11.897 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u500 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 20.78it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.945 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u250 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 20.78it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:11.984 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v1000 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 20.78it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 21.48it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.030 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v850 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 21.48it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.253 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v500 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 21.48it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.297 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v250 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 21.48it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 14.90it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.336 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z1000 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 14.90it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.391 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z850 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 14.90it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

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

                                                                                     
    
                                                                
    2025-03-27 08:03:12.441 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z500 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 15.77it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.487 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z250 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  62%|██████▏   | 16/26 [00:00<00:00, 15.77it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.527 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t1000 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  62%|██████▏   | 16/26 [00:01<00:00, 15.77it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

    Fetching GFS for 2025-03-26 00:00:00:  73%|███████▎  | 19/26 [00:01<00:00, 17.85it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.570 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t850 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  73%|███████▎  | 19/26 [00:01<00:00, 17.85it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.610 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t500 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  73%|███████▎  | 19/26 [00:01<00:00, 17.85it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.657 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t250 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  73%|███████▎  | 19/26 [00:01<00:00, 17.85it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

    Fetching GFS for 2025-03-26 00:00:00:  85%|████████▍ | 22/26 [00:01<00:00, 19.11it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.705 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q1000 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  85%|████████▍ | 22/26 [00:01<00:00, 19.11it/s]
    Running inference:  20%|██        | 1/5 [00:01<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.753 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q850 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  85%|████████▍ | 22/26 [00:01<00:00, 19.11it/s]
    Running inference:  20%|██        | 1/5 [00:02<00:03,  1.32it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.801 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q500 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  85%|████████▍ | 22/26 [00:01<00:00, 19.11it/s]
    Running inference:  20%|██        | 1/5 [00:02<00:03,  1.32it/s]

    Fetching GFS for 2025-03-26 00:00:00:  96%|█████████▌| 25/26 [00:01<00:00, 19.49it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:12.853 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q250 at 2025-03-26 00:00:00_0:00:00


    Fetching GFS for 2025-03-26 00:00:00:  96%|█████████▌| 25/26 [00:01<00:00, 19.49it/s]
    Running inference:  20%|██        | 1/5 [00:02<00:03,  1.32it/s]
    Fetching GFS for 2025-03-26 00:00:00: 100%|██████████| 26/26 [00:01<00:00, 18.63it/s]

    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]
                                                                
    2025-03-27 08:03:21.187 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:214 - Fetching GFS index file: 2025-03-26 00:00:00 lead 1:00:00

    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

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

                                                                            
    
                                                                
    2025-03-27 08:03:21.198 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u10m at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                            
    
                                                                
    2025-03-27 08:03:21.251 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v10m at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 18.05it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.309 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t2m at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 18.05it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.356 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: tcwv at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 18.05it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.397 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: msl at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 18.05it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.68it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.444 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: sp at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.68it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.487 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u1000 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.68it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.530 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u850 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.68it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

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

                                                                                    
    
                                                                
    2025-03-27 08:03:21.576 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u500 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 21.62it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.625 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u250 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 21.62it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:21.658 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v1000 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 21.62it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.66it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:21.700 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v850 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.66it/s]
    Running inference:  40%|████      | 2/5 [00:10<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:21.750 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v500 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.66it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:21.793 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v250 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.66it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 22.78it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:21.830 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z1000 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 22.78it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:21.878 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z850 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 22.78it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:21.926 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z500 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 22.78it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.55it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:21.966 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z250 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.55it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:22.006 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t1000 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.55it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:22.049 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t850 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.55it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

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

                                                                                     
    
                                                                
    2025-03-27 08:03:22.090 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t500 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 23.07it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:22.135 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t250 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 23.07it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:22.177 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q1000 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 23.07it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 22.66it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:22.227 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q850 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 22.66it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:22.272 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q500 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 22.66it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:22.311 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q250 at 2025-03-26 00:00:00_1:00:00


    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 22.66it/s]
    Running inference:  40%|████      | 2/5 [00:11<00:18,  6.02s/it]

    Fetching GFS for 2025-03-26 00:00:00: 100%|██████████| 26/26 [00:01<00:00, 22.42it/s]
    Fetching GFS for 2025-03-26 00:00:00: 100%|██████████| 26/26 [00:01<00:00, 22.29it/s]

    Running inference:  60%|██████    | 3/5 [00:19<00:15,  7.62s/it]
                                                                
    2025-03-27 08:03:30.720 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:214 - Fetching GFS index file: 2025-03-26 00:00:00 lead 2:00:00

    Running inference:  60%|██████    | 3/5 [00:19<00:15,  7.62s/it]

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

                                                                            
    
                                                                
    2025-03-27 08:03:30.735 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u10m at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                            
    
                                                                
    2025-03-27 08:03:30.799 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v10m at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 17.02it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:30.853 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t2m at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 17.02it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:30.886 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: tcwv at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 17.02it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:30.937 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: msl at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:   8%|▊         | 2/26 [00:00<00:01, 17.02it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.78it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:30.982 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: sp at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.78it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:31.025 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u1000 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.78it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:31.068 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u850 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  19%|█▉        | 5/26 [00:00<00:01, 20.78it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

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

                                                                                    
    
                                                                
    2025-03-27 08:03:31.108 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u500 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 22.17it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:31.146 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u250 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 22.17it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:31.184 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v1000 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  31%|███       | 8/26 [00:00<00:00, 22.17it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.98it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.232 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v850 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.98it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.275 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v500 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.98it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.310 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v250 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  42%|████▏     | 11/26 [00:00<00:00, 22.98it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 23.85it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.350 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z1000 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 23.85it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.402 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z850 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 23.85it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.460 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z500 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  54%|█████▍    | 14/26 [00:00<00:00, 23.85it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.38it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.500 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z250 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.38it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.536 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t1000 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.38it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.586 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t850 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  65%|██████▌   | 17/26 [00:00<00:00, 22.38it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

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

                                                                                     
    
                                                                
    2025-03-27 08:03:31.627 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t500 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 22.76it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.664 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t250 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 22.76it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.700 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q1000 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  77%|███████▋  | 20/26 [00:00<00:00, 22.76it/s]
    Running inference:  60%|██████    | 3/5 [00:20<00:15,  7.62s/it]

    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 23.15it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.752 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q850 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 23.15it/s]
    Running inference:  60%|██████    | 3/5 [00:21<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.803 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q500 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 23.15it/s]
    Running inference:  60%|██████    | 3/5 [00:21<00:15,  7.62s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:31.857 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q250 at 2025-03-26 00:00:00_2:00:00


    Fetching GFS for 2025-03-26 00:00:00:  88%|████████▊ | 23/26 [00:01<00:00, 23.15it/s]
    Running inference:  60%|██████    | 3/5 [00:21<00:15,  7.62s/it]

    Fetching GFS for 2025-03-26 00:00:00: 100%|██████████| 26/26 [00:01<00:00, 22.05it/s]
    Fetching GFS for 2025-03-26 00:00:00: 100%|██████████| 26/26 [00:01<00:00, 22.29it/s]

    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]
                                                                
    2025-03-27 08:03:40.263 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:214 - Fetching GFS index file: 2025-03-26 00:00:00 lead 3:00:00

    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

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

                                                                            
    
                                                                
    2025-03-27 08:03:40.274 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u10m at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

                                                                            
    
                                                                
    2025-03-27 08:03:40.326 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v10m at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

                                                                            
    
                                                                
    2025-03-27 08:03:40.373 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t2m at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:   0%|          | 0/26 [00:00<?, ?it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

    Fetching GFS for 2025-03-26 00:00:00:  12%|█▏        | 3/26 [00:00<00:01, 22.74it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.407 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: tcwv at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  12%|█▏        | 3/26 [00:00<00:01, 22.74it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.464 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: msl at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  12%|█▏        | 3/26 [00:00<00:01, 22.74it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.511 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: sp at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  12%|█▏        | 3/26 [00:00<00:01, 22.74it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

    Fetching GFS for 2025-03-26 00:00:00:  23%|██▎       | 6/26 [00:00<00:00, 20.94it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.558 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u1000 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  23%|██▎       | 6/26 [00:00<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.614 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u850 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  23%|██▎       | 6/26 [00:00<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.668 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u500 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  23%|██▎       | 6/26 [00:00<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

    Fetching GFS for 2025-03-26 00:00:00:  35%|███▍      | 9/26 [00:00<00:00, 20.89it/s]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.702 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: u250 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  35%|███▍      | 9/26 [00:00<00:00, 20.89it/s]
    Running inference:  80%|████████  | 4/5 [00:29<00:08,  8.38s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.744 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v1000 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  35%|███▍      | 9/26 [00:00<00:00, 20.89it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                    
    
                                                                
    2025-03-27 08:03:40.790 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v850 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  35%|███▍      | 9/26 [00:00<00:00, 20.89it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

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

                                                                                     
    
                                                                
    2025-03-27 08:03:40.843 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v500 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 21.03it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:40.877 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: v250 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 21.03it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:40.911 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z1000 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  46%|████▌     | 12/26 [00:00<00:00, 21.03it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

    Fetching GFS for 2025-03-26 00:00:00:  58%|█████▊    | 15/26 [00:00<00:00, 22.60it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:40.958 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z850 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  58%|█████▊    | 15/26 [00:00<00:00, 22.60it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.006 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z500 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  58%|█████▊    | 15/26 [00:00<00:00, 22.60it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.073 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: z250 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  58%|█████▊    | 15/26 [00:00<00:00, 22.60it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

    Fetching GFS for 2025-03-26 00:00:00:  69%|██████▉   | 18/26 [00:00<00:00, 20.94it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.122 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t1000 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  69%|██████▉   | 18/26 [00:00<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.168 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t850 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  69%|██████▉   | 18/26 [00:00<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.222 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t500 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  69%|██████▉   | 18/26 [00:00<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

    Fetching GFS for 2025-03-26 00:00:00:  81%|████████  | 21/26 [00:00<00:00, 20.94it/s]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.265 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: t250 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  81%|████████  | 21/26 [00:00<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.321 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q1000 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  81%|████████  | 21/26 [00:01<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.381 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q850 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  81%|████████  | 21/26 [00:01<00:00, 20.94it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

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

                                                                                     
    
                                                                
    2025-03-27 08:03:41.435 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q500 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  92%|█████████▏| 24/26 [00:01<00:00, 19.76it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]

                                                                                     
    
                                                                
    2025-03-27 08:03:41.487 | DEBUG    | earth2studio.data.gfs:_fetch_gfs_dataarray:260 - Fetching GFS grib file for variable: q250 at 2025-03-26 00:00:00_3:00:00


    Fetching GFS for 2025-03-26 00:00:00:  92%|█████████▏| 24/26 [00:01<00:00, 19.76it/s]
    Running inference:  80%|████████  | 4/5 [00:30<00:08,  8.38s/it]
    Fetching GFS for 2025-03-26 00:00:00: 100%|██████████| 26/26 [00:01<00:00, 20.64it/s]

    Running inference: 100%|██████████| 5/5 [00:39<00:00,  8.85s/it]
    Running inference: 100%|██████████| 5/5 [00:39<00:00,  7.84s/it]
    2025-03-27 08:03:49.938 | 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




.. GENERATED FROM PYTHON SOURCE LINES 111-118

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.

.. GENERATED FROM PYTHON SOURCE LINES 120-162

.. code-block:: Python

    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")



.. image-sg:: /examples/images/sphx_glr_09_stormcast_example_001.png
   :alt: 2025-03-26 - Lead time: 4hrs
   :srcset: /examples/images/sphx_glr_09_stormcast_example_001.png, /examples/images/sphx_glr_09_stormcast_example_001_2_00x.png 2.00x
   :class: sphx-glr-single-img






.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (2 minutes 56.951 seconds)


.. _sphx_glr_download_examples_09_stormcast_example.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: 09_stormcast_example.ipynb <09_stormcast_example.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: 09_stormcast_example.py <09_stormcast_example.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: 09_stormcast_example.zip <09_stormcast_example.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_