소스 검색

Check for NULL in ds column

bl 8 년 전
부모
커밋
8383d3bea6
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  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)