From 1be7bf7e5743e44b849f2c1663b2f258715cc691 Mon Sep 17 00:00:00 2001 From: braddf Date: Wed, 14 Feb 2024 10:41:11 +0000 Subject: [PATCH] Update pydantic types on dummy actuals --- src/india_api/internal/inputs/dummydb/client.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/src/india_api/internal/inputs/dummydb/client.py b/src/india_api/internal/inputs/dummydb/client.py index ea1ed7e..090bae7 100644 --- a/src/india_api/internal/inputs/dummydb/client.py +++ b/src/india_api/internal/inputs/dummydb/client.py @@ -67,18 +67,18 @@ def get_predicted_wind_yields_for_location( return values - def get_actual_solar_yields_for_location(self, location: str) -> list[internal.PredictedPower]: + def get_actual_solar_yields_for_location(self, location: str) -> list[internal.ActualPower]: """Gets the actual solar yields for a location.""" # Get the window start, end = get_window() numSteps = int((end - start) / step) - values: list[internal.PredictedPower] = [] + values: list[internal.ActualPower] = [] for i in range(numSteps): time = start + i * step _yield = _basicSolarYieldFunc(int(time.timestamp())) values.append( - internal.PredictedPower( + internal.ActualPower( Time=time, PowerKW=int(_yield.YieldKW), ), @@ -86,18 +86,18 @@ def get_actual_solar_yields_for_location(self, location: str) -> list[internal.P return values - def get_actual_wind_yields_for_location(self, location: str) -> list[internal.PredictedPower]: + def get_actual_wind_yields_for_location(self, location: str) -> list[internal.ActualPower]: """Gets the actual wind yields for a location.""" # Get the window start, end = get_window() numSteps = int((end - start) / step) - values: list[internal.PredictedPower] = [] + values: list[internal.ActualPower] = [] for i in range(numSteps): time = start + i * step _yield = _basicWindYieldFunc(int(time.timestamp())) values.append( - internal.PredictedPower( + internal.ActualPower( Time=time, PowerKW=int(_yield.YieldKW), ),