Explorar o código

Check for NULL in ds column

bl %!s(int64=8) %!d(string=hai) anos
pai
achega
8383d3bea6
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  1. 2 0
      python/fbprophet/forecaster.py

+ 2 - 0
python/fbprophet/forecaster.py

@@ -174,6 +174,8 @@ class Prophet(object):
         if 'y' in df:
             df['y'] = pd.to_numeric(df['y'])
         df['ds'] = pd.to_datetime(df['ds'])
+        if df['ds'].isnull().any():
+            raise ValueError('Found NaN in column ds.')
 
         df = df.sort_values('ds')
         df.reset_index(inplace=True, drop=True)