doppy 0.5.1__cp310-abi3-macosx_10_12_x86_64.whl → 0.5.3__cp310-abi3-macosx_10_12_x86_64.whl

doppy/product/turbulence.py

@@ -1,4 +1,3 @@
-# type: ignore
 from __future__ import annotations
 
 from dataclasses import dataclass
@@ -94,18 +93,18 @@ def _compute_variance(vert: VerticalWind, period: float) -> VarResult:
     N_cumsum = N_i.cumsum(axis=0)
 
     def N_func(i: int, j: int) -> npt.NDArray[np.float64]:
-        return N_cumsum[j] - N_cumsum[i] + N_i[i]
+        return np.array(N_cumsum[j] - N_cumsum[i] + N_i[i], dtype=np.float64)
 
     def S(i: int, j: int) -> npt.NDArray[np.float64]:
-        return X_cumsum[j] - X_cumsum[i] + X[i]
+        return np.array(X_cumsum[j] - X_cumsum[i] + X[i], dtype=np.float64)
 
     def S2(i: int, j: int) -> npt.NDArray[np.float64]:
-        return X2_cumsum[j] - X2_cumsum[i] + X2[i]
+        return np.array(X2_cumsum[j] - X2_cumsum[i] + X2[i], dtype=np.float64)
 
     def var_ij(i: int, j: int) -> npt.NDArray[np.float64]:
         N = N_func(i, j)
         with np.errstate(invalid="ignore"):
-            return (S2(i, j) - S(i, j) ** 2 / N) / N
+            return np.array((S2(i, j) - S(i, j) ** 2 / N) / N, dtype=np.float64)
 
     half_period = np.timedelta64(int(1e6 * period / 2), "us")
     period_start = np.full(vert.w.shape, np.datetime64("NaT", "us"))
@@ -147,7 +146,7 @@ def _length_scale_low(
 
 def _preprocess_horiontal_wind(
     vert: VerticalWind, hori: HorizontalWind, options: Options
-):
+) -> npt.NDArray[np.float64]:
     if np.isnan(hori.V).any():
         raise ValueError("horizontal wind speed cannot contains NaNs")
     trg_points = np.meshgrid(vert.time, vert.height, indexing="ij")
@@ -186,8 +185,8 @@ def _rolling_mean_over_time(
 
     S = arr.cumsum(axis=0)
 
-    def rolling_mean(i: int, j: int) -> npt.NDArray[np.flaat64]:
-        return (S[j] - S[i] + arr[i]) / (j - i + 1)
+    def rolling_mean(i: int, j: int) -> npt.NDArray[np.float64]:
+        return np.array((S[j] - S[i] + arr[i]) / (j - i + 1), dtype=np.float64)
 
     half_period = np.timedelta64(int(period * 0.5e6), "us")
     rol_mean = np.full(arr.shape, np.nan, dtype=np.float64)
@@ -208,7 +207,7 @@ def _compute_dissipation_rate(
     variance: npt.NDArray[np.float64],
     length_scale_lower: npt.NDArray[np.float64],
     length_scale_upper: npt.NDArray[np.float64],
-):
+) -> npt.NDArray[np.float64]:
     """
     Parameters
     ----------
@@ -225,10 +224,10 @@ def _compute_dissipation_rate(
             * (length_scale_upper ** (2 / 3) - length_scale_lower ** (2 / 3))
             ** (-3 / 2)
         )
-    return dr
+    return np.array(dr, dtype=np.float64)
 
 
-def _next_valid_from_mask(mask):
+def _next_valid_from_mask(mask: npt.NDArray[np.bool_]) -> npt.NDArray[np.int64]:
     """
     mask[t,v] (time,value)
 
@@ -244,10 +243,10 @@ def _next_valid_from_mask(mask):
     for t in reversed(range(n - 1)):
         N[t][~mask[t]] = t
         N[t][mask[t]] = N[t + 1][mask[t]]
-    return N
+    return np.array(N, dtype=np.int64)
 
 
-def _prev_valid_from_mask(mask):
+def _prev_valid_from_mask(mask: npt.NDArray[np.bool_]) -> npt.NDArray[np.int64]:
     """
     mask[t,v] (time,value)
 
@@ -262,4 +261,4 @@ def _prev_valid_from_mask(mask):
     for t in range(1, n):
         N[t][~mask[t]] = t
         N[t][mask[t]] = N[t - 1][mask[t]]
-    return N
+    return np.array(N, dtype=np.int64)

{doppy-0.5.1.dist-info → doppy-0.5.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: doppy
-Version: 0.5.1
+Version: 0.5.3
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10

{doppy-0.5.1.dist-info → doppy-0.5.3.dist-info}/RECORD

@@ -1,10 +1,10 @@
-doppy-0.5.1.dist-info/METADATA,sha256=Tv27VxO7NPbaTq0f5WQX2t06zSyi1yPouyVYcffzqXs,4315
-doppy-0.5.1.dist-info/WHEEL,sha256=VOPc4LnjY0yPhQSb3E0q8XFYhkjUnHylSjI09QJpAgM,105
-doppy-0.5.1.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
-doppy-0.5.1.dist-info/licenses/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
+doppy-0.5.3.dist-info/METADATA,sha256=iaWUuv-a0gsBRjMkHS05y9_YeS2sWph8ixuF43UAHFc,4315
+doppy-0.5.3.dist-info/WHEEL,sha256=VOPc4LnjY0yPhQSb3E0q8XFYhkjUnHylSjI09QJpAgM,105
+doppy-0.5.3.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
+doppy-0.5.3.dist-info/licenses/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
 doppy/options.py,sha256=73BDODO4OYHn2qOshhwz6u6G3J1kNd3uj6P0a3V4HBE,205
 doppy/__init__.py,sha256=Z9aEUlbPRWRUAoB8_-djkgrJuS4-6pjem4-mVSB6Z9I,191
-doppy/product/turbulence.py,sha256=AcLHQFeyBla0pP5CQtvJbYKupuuy-dwZK0CFlowBl9k,8251
+doppy/product/turbulence.py,sha256=hajTHFr6XXdul8oU9dw4Y58kBwRpUBIfiOfTcvSjq7M,8606
 doppy/product/wind.py,sha256=efBfYalILN2VvDiGpxP04mob-mCrdQFS9tdhReeyXiU,16787
 doppy/product/__init__.py,sha256=C6s9cX20m9UwRsKo1lZH6TnYFfM5KmsX5MPUyShbgl4,235
 doppy/product/stare_depol.py,sha256=gSlD13jGLRmHyTDADFs_MO6ECos6wwKF-TL7G7NwHAA,10718
@@ -28,5 +28,5 @@ doppy/raw/halo_bg.py,sha256=8t9j-SUF1yJht3vrT6KAYJQyxcg3W-0zr8h0jAEhWes,5815
 doppy/raw/halo_hpl.py,sha256=tK0H6LAOcoWqF4vvNu8VRCYq_spvcSTHRM4ju4gx5Lo,16828
 doppy/raw/windcube.py,sha256=HKAlU-TUVqA1sefRLOY5skM1twQaVe-MExlZoI40rK4,19149
 doppy/__main__.py,sha256=zrKQJVj0k0ypBQCGK65Czt9G9FZ_qx3ussw6Q9VJ14g,346
-doppy/rs.abi3.so,sha256=PNmTx9XM_g7_N-FzIJhn8ZTasGWo3u22mASKNwDYEgA,2716068
-doppy-0.5.1.dist-info/RECORD,,
+doppy/rs.abi3.so,sha256=v49dHuCKgYVheWZnDxI8jChnJHs7UJvQmHzbOdTFVvU,2677868
+doppy-0.5.3.dist-info/RECORD,,