DATA-2068/changing the logger method

Dataherald · Mar 26, 2024 · ecfa1f4 · ecfa1f4
1 parent f6472ef
commit ecfa1f4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dataherald/sql_generator/dataherald_finetuning_agent.py b/dataherald/sql_generator/dataherald_finetuning_agent.py
@@ -189,7 +189,7 @@ def similart_tables_based_on_few_shot_examples(self, df: pd.DataFrame) -> List[s
                 try:
                     tables = Parser(example["sql"]).tables
                 except Exception as e:
-                    logger.info(f"Error: {str(e)}")
+                    logger.error(f"Error parsing SQL: {str(e)}")
                 most_similar_tables.update(tables)
             df.drop(df[df.table_name.isin(most_similar_tables)].index, inplace=True)
         return most_similar_tables

diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py
@@ -250,7 +250,7 @@ def similart_tables_based_on_few_shot_examples(self, df: pd.DataFrame) -> List[s
                 try:
                     tables = Parser(example["sql"]).tables
                 except Exception as e:
-                    logger.info(f"Error: {str(e)}")
+                    logger.error(f"Error parsing SQL: {str(e)}")
                 most_similar_tables.update(tables)
             df.drop(df[df.table_name.isin(most_similar_tables)].index, inplace=True)
         return most_similar_tables