Skip to content

Commit a4664a1

Browse files
committed
use re for barckets space fix
1 parent 7d97f64 commit a4664a1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

eval/api_runner.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import requests
1414
from utils.reporting import upload_results
1515
import sqlparse
16+
import re
1617

1718

1819
def clean_generated_query(query: str):
@@ -34,9 +35,8 @@ def clean_generated_query(query: str):
3435
query = query.replace("/ NULLIF (", "/ NULLIF (1.0 * ")
3536

3637
# remove extra spaces around brackets especially for MySQL
37-
query = query.replace(" ( ", "(").replace(" )", ")")
38-
query = query.replace(" (", "(").replace(") ", ")")
39-
query = query.replace("( ", "(").replace(" )", ")")
38+
query = re.sub(r"\s*\(\s*", "(", query) # Remove spaces before and after '('
39+
query = re.sub(r"\s*\)", ")", query) # Remove spaces before ')'
4040

4141
return query
4242

0 commit comments

Comments
 (0)