Merge from main

ecmwf-lab · Jan 18, 2024 · 88ab5be · 88ab5be
1 parent 927fe4b
commit 88ab5be
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ai_models/inputs/__init__.py b/ai_models/inputs/__init__.py
@@ -26,7 +26,7 @@ def _patch(self, **kargs):
     def fields_sfc(self):
         param = self.owner.param_sfc
         if not param:
-            return cml.load_source("empty")
+            return ekd.load_source("empty")
 
         LOG.info(f"Loading surface fields from {self.WHERE}")
 
@@ -51,7 +51,7 @@ def fields_sfc(self):
     def fields_pl(self):
         param, level = self.owner.param_level_pl
         if not (param and level):
-            return cml.load_source("empty")
+            return ekd.load_source("empty")
 
         LOG.info(f"Loading pressure fields from {self.WHERE}")
         return ekd.from_source(
@@ -75,7 +75,7 @@ def fields_pl(self):
     def fields_ml(self):
         param, level = self.owner.param_level_ml
         if not (param and level):
-            return cml.load_source("empty")
+            return ekd.load_source("empty")
 
         LOG.info(f"Loading model fields from {self.WHERE}")
         return ekd.from_source(