WeatherNext2 Cyclone Mini

.agents/rules/e2s-013-assimilation-models.mdc

@@ -572,6 +572,6 @@ class MyDAModel(torch.nn.Module, AutoModelMixin):
 - **Register `device_buffer`** to track device via a `device` property
 - **Do NOT use `@batch_func` or `@batch_coords`** - these are px/dx conventions only
 - **Do NOT use `@torch.inference_mode()`** if the forward pass requires gradients (e.g., DPS guidance); document the reason if omitted
-- **Documentation**: Add the DA model to [models.rst](mdc:docs/modules/models.rst) in the `earth2studio.models.da` section, maintaining alphabetical order
+- **Documentation**: Add the DA model to [models_da.md](mdc:docs/modules/models_da.md), maintaining alphabetical order, and add an entry to [install_options.yml](mdc:docs/userguide/about/install_options.yml) when the model has an optional dependency extra or needs API install notes
 - DO NOT attempt to make a general base class with intent to reuse the wrapper
 - DO NOT over-populate the `load_model()` API - only expose essential parameters

CHANGELOG.md

@@ -11,6 +11,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+- Added WeatherNext 2 Cyclones Mini prognostic model wrapper (`WeatherNext2CyclonesMini`)
+
 ### Changed
 
 ### Deprecated

docs/modules/models_px.md

@@ -57,6 +57,7 @@ earth2studio.models.px.StormScopeGOES
 earth2studio.models.px.StormScopeMeteosatEU
 earth2studio.models.px.StormScopeMRMS
 earth2studio.models.px.UCast
+earth2studio.models.px.WeatherNext2CyclonesMini
 {% endautosummary %}
 
 <!-- mkdocs-badges:end -->

docs/userguide/about/install_options.yml

@@ -218,6 +218,20 @@ categories:
               - 'uv add "weathernext @ git+https://github.com/google-deepmind/weathernext.git@9c034db1ff412d5db6cbe6bb0c5c9afc5a267719"'
         notes:
           - GraphCast uses WeatherNext and requires Python 3.12 or newer.
+      - id: weathernext
+        label: WeatherNext 2 Cyclones Mini
+        extra: weathernext
+        summary: Google DeepMind WeatherNext 2 Cyclones Mini model dependencies.
+        api_refs:
+          - earth2studio.models.px.WeatherNext2CyclonesMini
+        preinstall:
+          pip:
+            - 'pip install "weathernext @ git+https://github.com/google-deepmind/weathernext.git@9c034db1ff412d5db6cbe6bb0c5c9afc5a267719"'
+          uv:
+            pypi:
+              - 'uv add "weathernext @ git+https://github.com/google-deepmind/weathernext.git@9c034db1ff412d5db6cbe6bb0c5c9afc5a267719"'
+        notes:
+          - WeatherNext 2 uses WeatherNext and requires Python 3.12 or newer.
       - id: interp-modafno
         label: InterpModAFNO
         extra: interp-modafno

earth2studio/lexicon/wb2.py

@@ -38,6 +38,8 @@ class WB2Lexicon(metaclass=LexiconType):
     VOCAB = {
         "u10m": "10m_u_component_of_wind::",
         "v10m": "10m_v_component_of_wind::",
+        "u100m": "100m_u_component_of_wind::",
+        "v100m": "100m_v_component_of_wind::",
         "t2m": "2m_temperature::",
         "sp": "surface_pressure::",
         "lsm": "land_sea_mask::",

earth2studio/models/px/__init__.py

@@ -55,6 +55,7 @@
 )
 from earth2studio.models.px.stormscope_meteosat import StormScopeMeteosatEU
 from earth2studio.models.px.ucast import UCast
+from earth2studio.models.px.weathernext2_cyclones_mini import WeatherNext2CyclonesMini
 
 # Silence warning spam from various models
 warnings.filterwarnings("ignore")