fix: forecast day init

WLM1ke · Jan 10, 2025 · eecf8ba · eecf8ba
1 parent 905b97f
commit eecf8ba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/poptimizer/domain/portfolio/forecasts.py b/poptimizer/domain/portfolio/forecasts.py
@@ -43,7 +43,8 @@ class Forecast(domain.Entity):
     mean: float = 0
     std: float = 0
 
-    def init_day(self) -> None:
+    def init_day(self, day: domain.Day) -> None:
+        self.day = day
         self.models.clear()
 
     def update_required(self, portfolio_ver: domain.Version) -> bool:

diff --git a/poptimizer/use_cases/portfolio/forecasts.py b/poptimizer/use_cases/portfolio/forecasts.py
@@ -24,7 +24,7 @@ async def __call__(
     ) -> handler.ForecastsAnalyzed | None:
         forecast = await ctx.get_for_update(forecasts.Forecast)
         if forecast.day < msg.day:
-            forecast.init_day()
+            forecast.init_day(msg.day)
 
         match msg:
             case handler.ModelDeleted():