Skip to content

Commit 802312e

Browse files
fix dataset (#1448)
1 parent c262e26 commit 802312e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/llm/utils/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1808,7 +1808,7 @@ def synthetic_text_to_sql_preprocesser(dataset: HfDataset):
18081808
def preprocess(row):
18091809
sql_prompt = row['sql_prompt']
18101810
sql_context = row['sql_context']
1811-
sql = row['sql_context']
1811+
sql = row['sql']
18121812
sql_explanation = row['sql_explanation']
18131813
query = f'Sql Table information:\n{sql_context}\n{sql_prompt}'
18141814
response = f'Let\'s think step by step:\n{sql_explanation}\nSo the final sql is:\n{sql}'

0 commit comments

Comments
 (0)