Hot fix forecasting (#1043)

georgia-tech-db · Sep 5, 2023 · 83a017e · 83a017e
1 parent e535896
commit 83a017e
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/evadb/binder/statement_binder.py b/evadb/binder/statement_binder.py
@@ -92,7 +92,7 @@ def _bind_create_function_statement(self, node: CreateFunctionStatement):
             inputs, outputs = [], []
             for column in all_column_list:
                 if column.name in predict_columns:
-                    if node.udf_type != "Forecasting":
+                    if node.function_type != "Forecasting":
                         column.name = column.name + "_predictions"
                     else:
                         column.name = column.name

diff --git a/evadb/udfs/forecast.py → evadb/functions/forecast.py b/evadb/udfs/forecast.py → evadb/functions/forecast.py
diff --git a/test/integration_tests/long/test_model_forecasting.py b/test/integration_tests/long/test_model_forecasting.py
@@ -51,7 +51,7 @@ def tearDownClass(cls):
     @forecast_skip_marker
     def test_forecast(self):
         create_predict_udf = """
-            CREATE UDF Forecast FROM
+            CREATE FUNCTION Forecast FROM
             (SELECT unique_id, ds, y FROM AirData)
             TYPE Forecasting
             PREDICT 'y';