force dtype in some numba case

Author: AntSimi

src/py_eddy_tracker/dataset/grid.py

@@ -100,7 +100,7 @@ def lat(self):
 def uniform_resample_stack(vertices, num_fac=2, fixed_size=None):
     x_val, y_val = vertices[:, 0], vertices[:, 1]
     x_new, y_new = uniform_resample(x_val, y_val, num_fac, fixed_size)
-    data = empty((x_new.shape[0], 2))
+    data = empty((x_new.shape[0], 2), dtype=vertices.dtype)
     data[:, 0] = x_new
     data[:, 1] = y_new
     return data

src/py_eddy_tracker/generic.py

@@ -120,7 +120,7 @@ def interp2d_geo(x_g, y_g, z_g, m_g, x, y):
     y_step = y_g[1] - y_ref
     nb_x = x_g.shape[0]
     is_circular = (x_g[-1] + x_step) % 360 == x_g[0] % 360
-    z = empty(x.shape)
+    z = empty(x.shape, dtype=z_g.dtype)
     for i in prange(x.size):
         x_ = (x[i] - x_ref) / x_step
         y_ = (y[i] - y_ref) / y_step

src/py_eddy_tracker/poly.py

@@ -115,8 +115,8 @@ def bbox_intersection(x0, y0, x1, y1):
     """
     nb0 = x0.shape[0]
     nb1 = x1.shape[0]
-    x1_min, y1_min = empty(nb1), empty(nb1)
-    x1_max, y1_max = empty(nb1), empty(nb1)
+    x1_min, y1_min = empty(nb1, dtype=x1.dtype), empty(nb1, dtype=x1.dtype)
+    x1_max, y1_max = empty(nb1, dtype=x1.dtype), empty(nb1, dtype=x1.dtype)
     for i1 in range(nb1):
         x1_min[i1], y1_min[i1] = x1[i1].min(), y1[i1].min()
         x1_max[i1], y1_max[i1] = x1[i1].max(), y1[i1].max()

tests/test_grid.py

@@ -7,6 +7,7 @@
 X = 0.025
 contour = Path(((-X, 0), (X, 0), (X, X), (-X, X), (-X, 0),))
 
+
 # contour
 def test_contour_lon():
     assert (contour.lon == (-X, X, X, -X, -X)).all()
@@ -43,4 +44,3 @@ def test_bounds():
     x0, x1, y0, y1 = G.bounds
     assert x0 == -1 / 120.0 and x1 == 360 - 1 / 120
     assert y0 == approx(-90 - 1 / 120.0) and y1 == approx(90 - 1 / 120)
-