doppy 0.5.1__cp310-abi3-win_amd64.whl → 0.5.2__cp310-abi3-win_amd64.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.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: doppy
-Version: 0.5.1
+Version: 0.5.2
 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.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
-doppy-0.5.1.dist-info/METADATA,sha256=wMmKn8B5USt6xoz3RGb4SOQyf7FqerfWMk5sas4si_M,4414
-doppy-0.5.1.dist-info/WHEEL,sha256=md9qofgGs0CN4-5nDhVd0IzxxzM8cFGCnkeRThUNGbI,95
-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=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
+doppy-0.5.2.dist-info/METADATA,sha256=mHDuliE31HGflXwmDeo0uCEscZHGj2e8iQ0r20nYKHI,4414
+doppy-0.5.2.dist-info/WHEEL,sha256=md9qofgGs0CN4-5nDhVd0IzxxzM8cFGCnkeRThUNGbI,95
+doppy-0.5.2.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
+doppy-0.5.2.dist-info/licenses/LICENSE,sha256=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
 doppy/bench.py,sha256=X4yPXFPAM708ptzXKDYxVBOC0E3cMj_KxXzmD_Q9_RM,321
 doppy/data/api.py,sha256=IoLRodfITNHs4Pb2IvVvOjSirSeK-xNoSO4tsQYjsKc,1933
 doppy/data/cache.py,sha256=ERFzH-KjLLXV8fCePsewTepKZXmLwtXxeazmHiFdA80,1203
@@ -14,7 +14,7 @@ doppy/options.py,sha256=uyIKM_G2GtbmV6Gve8o13eIShQqUwsnYZ41mhX2ypGE,218
 doppy/product/noise_utils.py,sha256=Bro3T8uMY9MmJcrXbOl_QKIU7OXVRUs_R6Norl5Tv3Q,2735
 doppy/product/stare.py,sha256=A_Xy6nIScnCUlEoziZ164aIW1HVqE7WogCcL5Ljpc_o,28009
 doppy/product/stare_depol.py,sha256=GkS-60Izk-rF9PnHPtOgAssr--K4ddzQHUlyLZ7JNkU,11026
-doppy/product/turbulence.py,sha256=IpqZ5ayxUrIX4J6C-3AcLTMem4OLHquKtvEC3zSEpkg,8516
+doppy/product/turbulence.py,sha256=RHvQlNobnXiF2Vx8OuvTYe7BIibmTNNXpHa7X4BfGbQ,8870
 doppy/product/wind.py,sha256=SbVM_AyLJtIggOb3PHLTak6majHaBMFJHNX36wRti1k,17298
 doppy/product/__init__.py,sha256=xoBEXuhid-bvoof5Ogzpt1dKIhJgNMRyrAinCqUOUUI,241
 doppy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,5 +28,5 @@ doppy/raw/__init__.py,sha256=ycT-_mCiI19b2TH33L74-Eq_MYUTPe3B8NhwNodSx1E,267
 doppy/utils.py,sha256=2PwDiO8GyMjsHC-EYcTGyN2mEfPaCi3hhEn2twETOl0,814
 doppy/__init__.py,sha256=Af7_8p3oN1nTqS9fo0mVKVuiKf5CAEK69uQa32CSFBA,197
 doppy/__main__.py,sha256=38hIWWfanILuBBGorQiAaleSC4qYJoIxuzVBkxf7Dng,371
-doppy/rs.pyd,sha256=iExSqsfsp1BACFKXzZBwf6DdFrQxOwnsDTZqbJC7yd8,2164736
-doppy-0.5.1.dist-info/RECORD,,
+doppy/rs.pyd,sha256=DDQzPRR6UTBJqDpmZfP9FL-1TJcg-26nSktIeVnVVww,2164736
+doppy-0.5.2.dist-info/RECORD,,