Make sure 'mu' and 'std' are not numpy floats (#2378)

This commit is contained in:
lior1990 2023-05-07 16:43:05 +03:00 committed by GitHub
parent 0bc1f9e784
commit 5e54831268
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -367,8 +367,8 @@ class Prophet(object):
else:
standardize = True
if standardize:
mu = df[name].mean()
std = df[name].std()
mu = float(df[name].mean())
std = float(df[name].std())
self.extra_regressors[name]['mu'] = mu
self.extra_regressors[name]['std'] = std