topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 12:37:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(improvement)(Headless) corrector supports subselect sql (#1006)

Browse Source 
* [improvement] corrector support subselect sql

---------

Co-authored-by: zuopengge
This commit is contained in:
mainmain
2024-05-17 10:01:49 +08:00
committed by GitHub
parent 2411cb3fdc
commit 7949efea75
6 changed files with 32 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
evaluation/build_pred_result.py
View File

@@ -66,6 +66,7 @@ def get_pred_result():
pred_sql_list=[]
default_sql="select * from tablea "
time_cost=[]
time.sleep(60)
for i in range(0,len(questions)):
start_time = time.time()
pred_sql=get_pred_sql(questions[i],url,agent_id,chat_id,authorization,default_sql)
@@ -73,7 +74,7 @@ def get_pred_result():
cost='%.3f'%(end_time-start_time)
time_cost.append(cost)
pred_sql_list.append(pred_sql)
time.sleep(60)
time.sleep(3)
write_sql(pred_sql_path, pred_sql_list)
return [float(cost) for cost in time_cost]