From 88211f3f9f77dbda97eb0d47e70e7ff3094d0eb7 Mon Sep 17 00:00:00 2001
From: xzdandy <xzdandy@gmail.com>
Date: Sat, 9 Sep 2023 15:03:12 -0400
Subject: [PATCH] Better way to specify column rename

---
 evadb/executor/create_function_executor.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/evadb/executor/create_function_executor.py b/evadb/executor/create_function_executor.py
index 5445fc8471..0b4aee75d7 100644
--- a/evadb/executor/create_function_executor.py
+++ b/evadb/executor/create_function_executor.py
@@ -225,9 +225,7 @@ def handle_forecasting_function(self):
         metadata_here = [
             FunctionMetadataCatalogEntry("model_name", model_name),
             FunctionMetadataCatalogEntry("model_path", model_path),
-            FunctionMetadataCatalogEntry(
-                "output_column_rename", self.node.outputs[0].name
-            ),
+            FunctionMetadataCatalogEntry("output_column_rename", arg_map["predict"]),
         ]
 
         return (