From 9e4e87af9baad421442a62206a6cb705ef4f612f Mon Sep 17 00:00:00 2001 From: Ben Letham Date: Wed, 3 Mar 2021 14:52:08 -0800 Subject: [PATCH] Make sure y_scale is not numpy float (#1824) --- python/fbprophet/forecaster.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/fbprophet/forecaster.py b/python/fbprophet/forecaster.py index 1f5c12e..40386c3 100644 --- a/python/fbprophet/forecaster.py +++ b/python/fbprophet/forecaster.py @@ -342,9 +342,9 @@ class Prophet(object): floor = df['floor'] else: floor = 0. - self.y_scale = (df['y'] - floor).abs().max() + self.y_scale = float((df['y'] - floor).abs().max()) if self.y_scale == 0: - self.y_scale = 1 + self.y_scale = 1.0 self.start = df['ds'].min() self.t_scale = df['ds'].max() - self.start for name, props in self.extra_regressors.items():