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, }, )