fix flat trend

NillionNetwork · Jun 11, 2024 · 56add95 · 56add95
1 parent 5b73d51
commit 56add95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nada_ai/time_series/prophet.py b/nada_ai/time_series/prophet.py
@@ -172,7 +172,7 @@ def predict_trend(self, floor: na.NadaArray, t: na.NadaArray) -> na.NadaArray:
             m_t = (-self.changepoints_t * deltas_t).sum(axis=1) + m
             trend = k_t * t + m_t
         elif self.growth == "flat":
-            trend = na.ones_like(t, na.rational) * m
+            trend = na.ones_like(t, na.Rational) * m
         else:
             raise NotImplementedError(self.growth + " is not supported")