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), ),