Commit
·
5bacb04
1
Parent(s):
95b85b7
chore: remove prints in talk to drias workflow
Browse files
climateqa/engine/talk_to_data/talk_to_drias.py
CHANGED
|
@@ -180,7 +180,6 @@ async def process_table(
|
|
| 180 |
table_state['status'] = 'ERROR'
|
| 181 |
return table_state
|
| 182 |
table_state['sql_query'] = sql_query
|
| 183 |
-
print(sql_query)
|
| 184 |
df = await execute_sql_query(sql_query)
|
| 185 |
|
| 186 |
table_state['dataframe'] = df
|
|
@@ -252,7 +251,6 @@ async def drias_workflow(user_input: str) -> State:
|
|
| 252 |
have_dataframe = False
|
| 253 |
have_sql_query = False
|
| 254 |
for table_state in results:
|
| 255 |
-
print(table_state)
|
| 256 |
if table_state['sql_query']:
|
| 257 |
have_sql_query = True
|
| 258 |
if table_state['dataframe'] is not None and len(table_state['dataframe']) > 0:
|
|
|
|
| 180 |
table_state['status'] = 'ERROR'
|
| 181 |
return table_state
|
| 182 |
table_state['sql_query'] = sql_query
|
|
|
|
| 183 |
df = await execute_sql_query(sql_query)
|
| 184 |
|
| 185 |
table_state['dataframe'] = df
|
|
|
|
| 251 |
have_dataframe = False
|
| 252 |
have_sql_query = False
|
| 253 |
for table_state in results:
|
|
|
|
| 254 |
if table_state['sql_query']:
|
| 255 |
have_sql_query = True
|
| 256 |
if table_state['dataframe'] is not None and len(table_state['dataframe']) > 0:
|