Fix completion plot for early finishes

matejak · Feb 21, 2024 · ea0f515 · ea0f515
1 parent c5d5cce
commit ea0f515
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/estimage/visualize/completion.py b/estimage/visualize/completion.py
@@ -18,7 +18,7 @@ def __init__(self, period_bounds, dom, cdf, ppf_cb):
         self.period_start = period_bounds[0]
         self.period_end = period_bounds[1]
         self.chart_days_before_dday = - min(int(self.dom[0]), (self.period_end - self.get_date_of_dday()).days)
-        self.chart_days_after_completion = min(0, (self.get_date_of_dday() + utils.ONE_DAY * self.dom[-1] - self.period_end).days)
+        self.chart_days_after_completion = - min(0, (self.get_date_of_dday() + utils.ONE_DAY * self.dom[-1] - self.period_end).days)
         self.ppf = ppf_cb
         self._pad_dom_and_cdf()
 
@@ -30,7 +30,7 @@ def _pad_dom_and_cdf(self):
         self.cdf = np.concatenate((
             np.zeros(self.chart_days_before_dday),
             self.cdf,
-            np.ones(self.chart_days_after_completion) * 100))
+            np.ones(self.chart_days_after_completion) * self.cdf[-1]))
 
     def _dom_to_days(self, dom_numbers):
         return dom_numbers - self.dom[0]