Better way to specify column rename

georgia-tech-db · Sep 9, 2023 · 88211f3 · 88211f3
1 parent 5a7b4dc
commit 88211f3
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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 (