Fix earthkit.geo import

ecmwf · Jun 3, 2024 · 38c0adc · 38c0adc
1 parent 1a1c6f4
commit 38c0adc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/anemoi/datasets/create/functions/filters/rotate_winds.py b/src/anemoi/datasets/create/functions/filters/rotate_winds.py
@@ -10,7 +10,7 @@
 from collections import defaultdict
 
 from earthkit.data.indexing.fieldlist import FieldArray
-from earthkit.geo.rotate import rotate_wind
+from earthkit.geo.rotate import rotate_vector
 
 
 class NewDataField:
@@ -65,7 +65,7 @@ def execute(
         assert x.grid_mapping == y.grid_mapping
 
         lats, lons = x.grid_points()
-        x_new, y_new = rotate_wind(
+        x_new, y_new = rotate_vector(
             lats,
             lons,
             x.to_numpy(flatten=True),

diff --git a/src/anemoi/datasets/create/functions/filters/unrotate_winds.py b/src/anemoi/datasets/create/functions/filters/unrotate_winds.py
@@ -11,7 +11,7 @@
 
 # import numpy as np
 from earthkit.data.indexing.fieldlist import FieldArray
-from earthkit.geo.rotate import unrotate_wind
+from earthkit.geo.rotate import unrotate_vector
 
 
 class NewDataField:
@@ -60,7 +60,7 @@ def execute(context, input, u, v):
 
         assert x.rotation == y.rotation
 
-        u_new, v_new = unrotate_wind(
+        u_new, v_new = unrotate_vector(
             lats,
             lons,
             x.to_numpy(flatten=True),