Merge pull request #250 from w-k-jones/fix_2d_latlon_interp_list

Fix 2d coordinate interpolation
tobac-project · Feb 23, 2023 · 819bba5 · 819bba5
2 parents 5ece4f5 + b78f1e0
commit 819bba5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tobac/utils/general.py b/tobac/utils/general.py
@@ -94,11 +94,15 @@ def add_coordinates(t, variable_cube):
 
             if variable_cube.coord_dims(coord) == (hdim_1, hdim_2):
                 f = interp2d(dimvec_2, dimvec_1, variable_cube.coord(coord).points)
-                coordinate_points = [f(a, b) for a, b in zip(t["hdim_2"], t["hdim_1"])]
+                coordinate_points = np.asarray(
+                    [f(a, b) for a, b in zip(t["hdim_2"], t["hdim_1"])]
+                )
 
             if variable_cube.coord_dims(coord) == (hdim_2, hdim_1):
                 f = interp2d(dimvec_1, dimvec_2, variable_cube.coord(coord).points)
-                coordinate_points = [f(a, b) for a, b in zip(t["hdim_1"], t["hdim_2"])]
+                coordinate_points = np.asarray(
+                    [f(a, b) for a, b in zip(t["hdim_1"], t["hdim_2"])]
+                )
 
         # interpolate 3D coordinates:
         # mainly workaround for wrf latitude and longitude (to be fixed in future)