update test

georgia-tech-db · Nov 17, 2023 · 2d38124 · 2d38124
1 parent 8d2d5ef
commit 2d38124
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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",
             ],
         )