table params

evaluation_logic.py

@@ -63,18 +63,10 @@ def run_prediction(model_name, prompt_format, output_file):
             if len(table_params) == 0:
                 yield f"[red] WARNING: No tables found for {db_id} [/red]"
 
-            # Convert Table objects to dictionaries if they're already instantiated
-            processed_table_params = []
-            for table in table_params:
-                if isinstance(table, Table):
-                    processed_table_params.append(table.dict())
-                else:
-                    processed_table_params.append(table)
-
             text_to_sql_inputs.append(TextToSQLParams(
                 instruction=question,
                 database=db_id,
-                tables=processed_table_params,
+                tables=table_params,
             ))
 
         # Generate SQL