From 1006c02d96a4290d504a093efd35bd2ff494ecbe Mon Sep 17 00:00:00 2001 From: devsjc <47188100+devsjc@users.noreply.github.com> Date: Mon, 23 Mar 2026 08:55:34 +0000 Subject: [PATCH] fix(modatahub): Add pressure to nl model --- src/nwp_consumer/internal/entities/modelmetadata.py | 10 ++++++++++ .../repositories/raw_repositories/mo_datahub.py | 4 +++- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/src/nwp_consumer/internal/entities/modelmetadata.py b/src/nwp_consumer/internal/entities/modelmetadata.py index 03d5482f..b537e859 100644 --- a/src/nwp_consumer/internal/entities/modelmetadata.py +++ b/src/nwp_consumer/internal/entities/modelmetadata.py @@ -211,6 +211,16 @@ def with_delay_minutes(self, delay_minutes: int) -> "ModelMetadata": """Returns metadata for the given model with the given delay minutes.""" return dataclasses.replace(self, delay_minutes=delay_minutes) + def with_extra_parameters(self, parameters: list[Parameter]) -> "ModelMetadata": + """Returns metadata for the given model with the given extra parameters.""" + return dataclasses.replace( + self, + expected_coordinates=dataclasses.replace( + self.expected_coordinates, + variable=sorted(self.expected_coordinates.variable + parameters), + ), + ) + def with_max_step(self, max_step: int) -> "ModelMetadata": """Returns metadata for the given model with the given max step.""" return dataclasses.replace( diff --git a/src/nwp_consumer/internal/repositories/raw_repositories/mo_datahub.py b/src/nwp_consumer/internal/repositories/raw_repositories/mo_datahub.py index a8e8eb01..2bd31a64 100644 --- a/src/nwp_consumer/internal/repositories/raw_repositories/mo_datahub.py +++ b/src/nwp_consumer/internal/repositories/raw_repositories/mo_datahub.py @@ -165,7 +165,9 @@ def repository() -> entities.RawRepositoryMetadata: "default": entities.Models.MO_UM_GLOBAL_10KM.with_region("india"), "um-global-10km-india": entities.Models.MO_UM_GLOBAL_10KM.with_region("india"), "um-global-10km-uk": entities.Models.MO_UM_GLOBAL_10KM.with_region("uk"), - "um-global-10km-nl": entities.Models.MO_UM_GLOBAL_10KM.with_region("nl"), + "um-global-10km-nl": entities.Models.MO_UM_GLOBAL_10KM\ + .with_region("nl")\ + .with_extra_parameters([entities.Parameter.PRESSURE_MSL]), "um-ukv-2km": entities.Models.MO_UM_UKV_2KM_LAEA, }, )