Update Support_module_dir/support_function_predict.py

Support_module_dir/support_function_predict.py

@@ -1,5 +1,6 @@
 import time
 import warnings
+from Variable_artifects.artifact import FORECAST_DAY
 import pandas as pd
 
 
@@ -15,7 +16,7 @@ def predict_function(trained_model, dataset, forecast_days):
 
     Returns
     -------
-    predict_frame : pandas Dataframe with predicted values and corresponding dates.
+    predict_frame : pandas Dataframe  with predicted values and corresponding dates.
     execution_time : float
         Execution time for making predictions.
     """
@@ -45,8 +46,10 @@ def predict_function(trained_model, dataset, forecast_days):
             last_date = dataset.index[-2]
 
             # Generate a new DataFrame with the dates for the next x number of steps
-            new_dates = pd.date_range(start=last_date, periods=PERIODS, freq='B')[
-                        1:]  # 'B' stands for business day), [1:] Exclude the last_date
+            new_dates = pd.date_range(
+                start=last_date,
+                periods=PERIODS,
+                freq='B')[1:]  # 'B' stands for business day), [1:] Exclude the last_date
 
             # Print or use the confidence interval
             predict_frame = pd.DataFrame({"Date": new_dates, "Lower_Bound": confidence_interval.iloc[:, 0],