Merge pull request #52 from ecmwf-lab/bugfix/retrieve-request-ml

Added request info for ML
ecmwf-lab · Jul 29, 2024 · af35dd8 · af35dd8
2 parents 590e438 + dfe47ef
commit af35dd8
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/ai_models/model.py b/src/ai_models/model.py
@@ -374,6 +374,7 @@ def _requests_unfiltered(self):
         for date, time in self.datetimes():  # noqa F402
             param, level = self.param_level_pl
 
+            # PL
             r = dict(
                 levtype="pl",
                 levelist=level,
@@ -390,6 +391,25 @@ def _requests_unfiltered(self):
 
             result.append(dict(**r))
 
+            # ML
+            param, level = self.param_level_ml
+            r = dict(
+                levtype="ml",
+                levelist=level,
+                param=param,
+                date=date,
+                time=time,
+            )
+            r.update(first)
+            first = {}
+
+            r.update(self._requests_extra)
+
+            self.patch_retrieve_request(r)
+
+            result.append(dict(**r))
+
+            # SFC
             r.update(
                 dict(
                     levtype="sfc",