.. DO NOT EDIT. .. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY. .. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE: .. "examples/02_diagnostic_workflow.py" .. LINE NUMBERS ARE GIVEN BELOW. .. only:: html .. note:: :class: sphx-glr-download-link-note :ref:`Go to the end ` to download the full example code. .. rst-class:: sphx-glr-example-title .. _sphx_glr_examples_02_diagnostic_workflow.py: Running Diagnostic Inference ============================ Basic prognostic + diagnostic inference workflow. This example will demonstrate how to run a deterministic inference workflow that couples a prognostic model with a diagnostic model. This diagnostic model will predict a new atmospheric quantity from the predicted fields of the prognostic. In this example you will learn: - How to instantiate a prognostic model - How to instantiate a diagnostic model - Creating a data source and IO object - Running the built in diagnostic workflow - Post-processing results .. GENERATED FROM PYTHON SOURCE LINES 38-42 Set Up ------ For this example, the built in diagnostic workflow :py:meth:`earth2studio.run.diagnostic` will be used. .. GENERATED FROM PYTHON SOURCE LINES 44-48 .. literalinclude:: ../../earth2studio/run.py :language: python :start-after: # sphinx - diagnostic start :end-before: # sphinx - diagnostic end .. GENERATED FROM PYTHON SOURCE LINES 51-57 Thus, we need the following: - Prognostic Model: Use the built in FourCastNet Model :py:class:`earth2studio.models.px.FCN`. - Diagnostic Model: Use the built in precipitation AFNO model :py:class:`earth2studio.models.dx.PrecipitationAFNO`. - Datasource: Pull data from the GFS data api :py:class:`earth2studio.data.GFS`. - IO Backend: Save the outputs into a Zarr store :py:class:`earth2studio.io.ZarrBackend`. .. GENERATED FROM PYTHON SOURCE LINES 59-84 .. code-block:: Python 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 from earth2studio.io import ZarrBackend from earth2studio.models.dx import PrecipitationAFNO from earth2studio.models.px import FCN # Load the default model package which downloads the check point from NGC package = FCN.load_default_package() prognostic_model = FCN.load_model(package) package = PrecipitationAFNO.load_default_package() diagnostic_model = PrecipitationAFNO.load_model(package) # Create the data source data = GFS() # Create the IO handler, store in memory io = ZarrBackend() .. rst-class:: sphx-glr-script-out .. code-block:: none /usr/lib/python3.12/tarfile.py:2254: DeprecationWarning: Python 3.14 will, by default, filter extracted tar archives and reject files or modify their metadata. Use the filter argument to control this behavior. warnings.warn( .. GENERATED FROM PYTHON SOURCE LINES 85-91 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. .. GENERATED FROM PYTHON SOURCE LINES 93-102 .. code-block:: Python import earth2studio.run as run nsteps = 8 io = run.diagnostic( ["2021-06-01"], nsteps, prognostic_model, diagnostic_model, data, io ) print(io.root.tree()) .. rst-class:: sphx-glr-script-out .. code-block:: none 2025-03-27 07:56:32.050 | INFO | earth2studio.run:diagnostic:190 - Running diagnostic workflow! 2025-03-27 07:56:32.050 | INFO | earth2studio.run:diagnostic:197 - Inference device: cuda /home/jenkins/agent/workspace/Studio-Build-Docs/repo/earth2studio/utils/time.py:38: DeprecationWarning: datetime.datetime.utcfromtimestamp() is deprecated and scheduled for removal in a future version. Use timezone-aware objects to represent datetimes in UTC: datetime.datetime.fromtimestamp(timestamp, datetime.UTC). time = [datetime.utcfromtimestamp((date - _unix) / _ds) for date in time] 2025-03-27 07:56:32.110 | DEBUG | earth2studio.data.gfs:_fetch_gfs_dataarray:214 - Fetching GFS index file: 2021-06-01 00:00:00 lead 0:00:00 Fetching GFS for 2021-06-01 00:00:00: 0%| | 0/26 [00:00` .. container:: sphx-glr-download sphx-glr-download-python :download:`Download Python source code: 02_diagnostic_workflow.py <02_diagnostic_workflow.py>` .. container:: sphx-glr-download sphx-glr-download-zip :download:`Download zipped: 02_diagnostic_workflow.zip <02_diagnostic_workflow.zip>` .. only:: html .. rst-class:: sphx-glr-signature `Gallery generated by Sphinx-Gallery `_