Add Aurora v1.5 deterministic and ensemble prognostic model wrappers

CHANGELOG.md

@@ -11,6 +11,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+- Added Aurora v1.5 deterministic and ensemble model wrapper (`Aurora1p5`, `Aurora1p5Ensemble`)
 - Added EarthMover ERA5 0.25 degree reanalysis data source
 - Added EarthMover IFS 0.1 degree data source and forecast source hosted by BrightBand
 

docs/modules/models_px.rst

@@ -40,6 +40,8 @@ Thus are typically used to generate forecast predictions.
       AIFSENS
       Atlas
       Aurora
+      Aurora1p5
+      Aurora1p5Ensemble
       CBottleVideo
       DiagnosticWrapper
       DLESyM

docs/userguide/about/install.md

@@ -224,7 +224,28 @@ uv add earth2studio --extra atlas
 ::::
 :::::
 :::::{tab-item} Aurora
-Notes: The Aurora model relies on the [microsoft aurora](https://github.com/microsoft/aurora)
+Notes: The Aurora model relies on the [Microsoft Aurora](https://github.com/microsoft/aurora)
+package for inference.
+
+::::{tab-set}
+:::{tab-item} pip
+
+```bash
+pip install earth2studio[aurora]
+```
+
+:::
+:::{tab-item} uv
+
+```bash
+uv add earth2studio --extra aurora
+```
+
+:::
+::::
+:::::
+:::::{tab-item} Aurora v1.5
+Notes: The Aurora v1.5 model relies on the [Microsoft Aurora](https://github.com/microsoft/aurora)
 package for inference.
 
 ::::{tab-set}

earth2studio/lexicon/base.py

@@ -77,6 +77,7 @@ def __contains__(cls, val: object) -> bool:
     "sd": "snow depth (m of water equivalent)",
     "sde": "snow depth (m)",
     "fg10m": "maximum 10 m wind gust since previous post-processing (m s-1)",
+    "i10fg": "instantaneous 10 m wind gust (m s-1)",
     "slhf": "time-integrated surface latent heat net flux (J m-2)",
     "ssr": "surface net short-wave radiation (J m-2)",
     "rlut": "outgoing longwave radiation (W m-2)",
@@ -358,8 +359,13 @@ def __contains__(cls, val: object) -> bool:
     "ssrd": "surface short-wave (solar) radiation downwards (J m-2) accumulated over a time period",
     "strd": "surface long-wave (thermal) radiation downwards (J m-2) accumulated over a time period",
     "ssrd06": "surface short-wave (solar) radiation downwards (J m-2) past 6 hours",
+    "ssrd1h": "surface short-wave (solar) radiation downwards (J m-2) past 1 hour",
     "strd06": "surface long-wave (thermal) radiation downwards (J m-2) past 6 hours",
+    "ttr1h": "top net thermal (longwave) radiation (J m-2) past 1 hour",
+    "uvb1h": "UV biologically effective dose (J m-2) past 1 hour",
+    "tp1h": "total precipitation (m) past 1 hour",
     "sf": "snowfall water equivalent (kg m-2)",
+    "sf1h": "snowfall (m of water equivalent) past 1 hour",
     "ro": "runoff water equivalent (surface plus subsurface) (kg m-2)",
     "swh": "significant wave height of combined wind waves and swell (m)",
     "mwd": "mean wave direction (degrees true)",

earth2studio/lexicon/ncar.py

@@ -56,10 +56,15 @@ def build_vocab() -> dict[str, str]:
             "t2m": (167, "2t", "e5.oper.an.sfc", "ll025sc"),
             "u100m": (246, "100u", "e5.oper.an.sfc", "ll025sc"),
             "v100m": (247, "100v", "e5.oper.an.sfc", "ll025sc"),
+            "sic": (31, "ci", "e5.oper.an.sfc", "ll025sc"),
             "skt": (235, "skt", "e5.oper.an.sfc", "ll025sc"),
             "stl1": (139, "stl1", "e5.oper.an.sfc", "ll025sc"),
             "stl2": (170, "stl2", "e5.oper.an.sfc", "ll025sc"),
             "sd": (141, "sd", "e5.oper.an.sfc", "ll025sc"),
+            "tcc": (164, "tcc", "e5.oper.an.sfc", "ll025sc"),
+            "lcc": (186, "lcc", "e5.oper.an.sfc", "ll025sc"),
+            "mcc": (187, "mcc", "e5.oper.an.sfc", "ll025sc"),
+            "hcc": (188, "hcc", "e5.oper.an.sfc", "ll025sc"),
             "lsp": (142, "lsp", "e5.oper.fc.sfc.accumu", "ll025sc"),  # hourly (m)
             "cp": (143, "cp", "e5.oper.fc.sfc.accumu", "ll025sc"),  # hourly (m)
             "smlt": (45, "smlt", "e5.oper.fc.sfc.accumu", "ll025sc"),  # hourly (m)

earth2studio/models/px/__init__.py

@@ -23,6 +23,7 @@
 from earth2studio.models.px.aifsens import AIFSENS
 from earth2studio.models.px.atlas import Atlas
 from earth2studio.models.px.aurora import Aurora
+from earth2studio.models.px.aurora1p5 import Aurora1p5, Aurora1p5Ensemble
 from earth2studio.models.px.base import PrognosticModel
 from earth2studio.models.px.cbottle_video import CBottleVideo
 from earth2studio.models.px.dlesym import DLESyM, DLESyMLatLon