wip: fix test

georgia-tech-db · Sep 25, 2023 · b7e70f2 · b7e70f2
1 parent 52c563e
commit b7e70f2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/integration_tests/long/test_model_forecasting.py b/test/integration_tests/long/test_model_forecasting.py
@@ -58,7 +58,7 @@ def setUpClass(cls):
         load_query = f"LOAD CSV '{path}' INTO AirData;"
         execute_query_fetch_all(cls.evadb, load_query)
 
-        path = f"{EvaDB_ROOT_DIR}/data/forecasting/ AirPassengersPanel.csv"
+        path = f"{EvaDB_ROOT_DIR}/data/forecasting/AirPassengersPanel.csv"
         load_query = f"LOAD CSV '{path}' INTO AirDataPanel;"
         execute_query_fetch_all(cls.evadb, load_query)
 
@@ -82,8 +82,8 @@ def test_forecast(self):
         create_predict_udf = """
             CREATE FUNCTION AirForecast FROM
             (SELECT unique_id, ds, y FROM AirData)
-            HORIZON 12
             TYPE Forecasting
+            HORIZON 12
             PREDICT 'y';
         """
         execute_query_fetch_all(self.evadb, create_predict_udf)
@@ -100,8 +100,8 @@ def test_forecast(self):
         create_predict_udf = """
             CREATE FUNCTION AirPanelForecast FROM
             (SELECT unique_id, ds, y, trend FROM AirDataPanel)
-            HORIZON 12
             TYPE Forecasting
+            HORIZON 12
             PREDICT 'y'
             LIBRARY 'neuralforecast'
             EXOGENOUS 'trend'