Skip to content

Commit b280204

Browse files
DH-5688/reformat with black
1 parent c97f66f commit b280204

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

dataherald/sql_generator/dataherald_finetuning_agent.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -227,9 +227,7 @@ def _run(
227227
most_similar_tables = self.similart_tables_based_on_few_shot_examples(df)
228228
table_relevance = ""
229229
for _, row in df.iterrows():
230-
table_relevance += (
231-
f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'
232-
)
230+
table_relevance += f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'
233231
if len(most_similar_tables) > 0:
234232
for table in most_similar_tables:
235233
table_relevance += (

dataherald/sql_generator/dataherald_sqlagent.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -290,9 +290,7 @@ def _run(
290290
most_similar_tables = self.similart_tables_based_on_few_shot_examples(df)
291291
table_relevance = ""
292292
for _, row in df.iterrows():
293-
table_relevance += (
294-
f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'
295-
)
293+
table_relevance += f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'
296294
if len(most_similar_tables) > 0:
297295
for table in most_similar_tables:
298296
table_relevance += (

0 commit comments

Comments
 (0)