Make mypy happy

eWaterCycle · Jun 20, 2022 · b9b3fae · b9b3fae
1 parent 8013b68
commit b9b3fae
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/ewatercycle/models/hype.py b/src/ewatercycle/models/hype.py
@@ -202,7 +202,7 @@ def _coords_to_indices(
         for plon, plat in zip(lon, lat):
             dist = geographical_distances(plon, plat, x, y)
             index = dist.argmin()
-            indices.append(index)
+            indices.append(int(index))
 
         return indices
 

diff --git a/src/ewatercycle/util.py b/src/ewatercycle/util.py
@@ -46,16 +46,16 @@ def find_closest_point(
     if distance > max(dx, dy) * 2:
         raise ValueError(f"Point {point_longitude, point_latitude} outside model grid.")
 
-    return idx_lon, idx_lat
+    return int(idx_lon), int(idx_lat)
 
 
 def geographical_distances(
     point_longitude: float,
     point_latitude: float,
-    lon_vectors: np.array,
-    lat_vectors: np.array,
+    lon_vectors: np.ndarray,
+    lat_vectors: np.ndarray,
     radius=6373.0,
-) -> np.array:
+) -> np.ndarray:
     """It uses Spherical Earth projected to a plane formula:
     https://en.wikipedia.org/wiki/Geographical_distance