From 2d3812438f03e4fda1bd558cd1779c25e5af12ba Mon Sep 17 00:00:00 2001
From: americast <sayan.sinha@cc.gatech.edu>
Date: Fri, 17 Nov 2023 01:39:18 -0500
Subject: [PATCH] update test

---
 test/integration_tests/long/test_model_forecasting.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/integration_tests/long/test_model_forecasting.py b/test/integration_tests/long/test_model_forecasting.py
index c376f7610c..f883de221d 100644
--- a/test/integration_tests/long/test_model_forecasting.py
+++ b/test/integration_tests/long/test_model_forecasting.py
@@ -101,6 +101,7 @@ def test_forecast(self):
                 "airforecast.y",
                 "airforecast.y-lo",
                 "airforecast.y-hi",
+                "plot",
             ],
         )
 
@@ -122,7 +123,7 @@ def test_forecast_neuralforecast(self):
         execute_query_fetch_all(self.evadb, create_predict_udf)
 
         predict_query = """
-            SELECT AirPanelForecast() order by y;
+            SELECT AirPanelForecast(FALSE) order by y;
         """
         result = execute_query_fetch_all(self.evadb, predict_query)
         self.assertEqual(len(result), 24)
@@ -167,6 +168,7 @@ def test_forecast_with_column_rename(self):
                 "homeforecast.ma",
                 "homeforecast.ma-lo",
                 "homeforecast.ma-hi",
+                "plot",
             ],
         )