use dicts

ecmwf · Jan 16, 2025 · f587698 · f587698
1 parent 5e61b0d
commit f587698
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/anemoi/transform/filters/regrid.py b/src/anemoi/transform/filters/regrid.py
@@ -9,6 +9,7 @@
 
 
 import logging
+from collections import defaultdict
 
 from earthkit.data.core.fieldlist import Field
 
@@ -127,7 +128,6 @@ def __call__(self, field):
 
 
 def interpolator(in_grid, out_grid, method):
-    if method.startswith("anemoi."):
-        method = method.split(".")[1]
-        return AnemoiInterpolator(in_grid, out_grid, method)
-    return EarthkitInterpolator(in_grid, out_grid, method)
+    INTERPOLATORS = defaultdict(lambda: EarthkitInterpolator)
+    INTERPOLATORS["nearest"] = AnemoiInterpolator
+    return INTERPOLATORS[method](in_grid, out_grid, method)