domainiac 9.0.3__tar.gz → 9.0.5__tar.gz

{domainiac-9.0.3 → domainiac-9.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: domainiac
-Version: 9.0.3
+Version: 9.0.5
 Summary: Package for working with Energinet data, but with specialized functions used for Enigma.
 Author: Team Enigma
 Author-email: gridop-enigma@energinet.dk

{domainiac-9.0.3 → domainiac-9.0.5}/domainiac/functions/interpolation.py

@@ -112,6 +112,7 @@ def binned_index_interpolation(
     index = interpolate.PchipInterpolator(
         x=bin_midpoints,
         y=bin_index,
+        extrapolate=False,
     )
 
     def estimate(x: NDArray[np.float64]) -> NDArray[np.float64]:

{domainiac-9.0.3 → domainiac-9.0.5}/domainiac/functions/temperature.py

@@ -23,8 +23,12 @@ def interpolate_temperature(times: pd.Series, temperature: pd.Series) -> RealFun
 
     f = interpolate.make_interp_spline(x, y, k=1)
 
+    bounds = (x.min(), x.max())
+
     def estimate(times: pd.Series) -> NDArray[np.float64]:
         x = conversions.datetime_to_float(times)
-        return f(x)
+        y = f(x)
+        y[(x < bounds[0]) | (x > bounds[1])] = np.nan
+        return y
 
     return estimate

{domainiac-9.0.3 → domainiac-9.0.5}/domainiac/functions/wind.py

@@ -25,8 +25,12 @@ def interpolate_wind_components(
 
     f = interpolate.make_interp_spline(x, y, k=1)
 
+    bounds = (x.min(), x.max())
+
     def estimate(times: pd.Series) -> NDArray[np.float64]:
         x = conversions.datetime_to_float(times)
-        return f(x)
+        y = f(x)
+        y[(x < bounds[0]) | (x > bounds[1]), :] = np.nan
+        return y
 
     return estimate

{domainiac-9.0.3 → domainiac-9.0.5}/domainiac/managers/nwp_manager.py

@@ -51,9 +51,10 @@ class NWPManager:
         """
         key = (provider, parameter)
         if key not in self._nwp_coordinates_kd_tree:
-            # Base KDTree on one hour of data
+            # Base KDTree on six hours of data,
+            # which should be enough to cover the largest interval which is ECMWF
             query_time_interval = pdz.TimeInterval(
-                self.time_interval.left, self.time_interval.left + pd.Timedelta("PT1H")
+                self.time_interval.left, self.time_interval.left + pd.Timedelta("PT6H")
             )
 
             df = self.db.query(
@@ -194,7 +195,11 @@ class NWPManager:
             df_interpolated["time_utc"]
         )
 
-        # disallow extrapolation
+        # remove nan values from the interpolated data
+        # (for example when data has been extrapolated)
+        df = df_interpolated.dropna(subset=column_names)
+
+        # remove values outside the original time interval
         df_interpolated = df_interpolated[
             df_interpolated["time_utc"].between(
                 df["time_utc"].min(), df["time_utc"].max()

{domainiac-9.0.3 → domainiac-9.0.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "domainiac"
-version = "9.0.3"
+version = "9.0.5"
 description = "Package for working with Energinet data, but with specialized functions used for Enigma."
 authors = ["Team Enigma <gridop-enigma@energinet.dk>"]
 packages = [