v1.1 release fix

app.py

@@ -124,8 +124,7 @@ def get_forecast_period(period):
 
 # SARIMAX Model
 @st.cache_data
-def train_test(dataframe):
-  n = round(len(dataframe) * 0.2)
+def train_test(dataframe, n):
   training_y = dataframe.iloc[:-n,0]
   test_y = dataframe.iloc[-n:,0]
   test_y_series = pd.Series(test_y, index=dataframe.iloc[-n:, 0].index)
@@ -303,8 +302,6 @@ with st.sidebar:
       df = date_format(df)
       merge_sort(df)
       series = group_to_three(df)
-      print(series) # debug
-      print(len(series)) # debug
 
       st.session_state.uploaded = True
 
@@ -328,13 +325,14 @@ if (st.session_state.uploaded):
 
   if (forecast_button or st.session_state.forecasted):
     df = series_to_df_exogenous(series)
+    print(df) # debug
+    print(len(df)) # debug
+    n_periods = round(len(df) * 0.2)
 
-    train = train_test(df)
+    train = train_test(df, n_periods)
     training_y, test_y, test_y_series, training_X, test_X, future_X = train
     train_test_model = test_fitting(df, training_X, training_y)
     
-    n_periods = round(len(df) * 0.2)
-    print(n_periods) # debug
     future_n_periods = forecast_period
     fitted, confint = train_test_model.predict(X=test_X, n_periods=n_periods, return_conf_int=True)
     index_of_fc = test_y_series.index