Skip to content

Commit

Permalink
DATA-5688/fix the inline comments
Browse files Browse the repository at this point in the history
  • Loading branch information
MohammadrezaPourreza committed Apr 4, 2024
1 parent 19786a0 commit c97f66f
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions dataherald/sql_generator/dataherald_finetuning_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -228,12 +228,12 @@ def _run(
table_relevance = ""
for _, row in df.iterrows():
table_relevance += (
f'Table: {row["table_name"]}, relevance score: {row["similarities"]}\n'
f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'
)
if len(most_similar_tables) > 0:
for table in most_similar_tables:
table_relevance += (
f"Table: {table}, relevance score: {max(df['similarities'])}\n"
f"Table: `{table}`, relevance score: {max(df['similarities'])}\n"
)
return table_relevance

Expand Down
4 changes: 2 additions & 2 deletions dataherald/sql_generator/dataherald_sqlagent.py
Original file line number Diff line number Diff line change
Expand Up @@ -291,12 +291,12 @@ def _run(
table_relevance = ""
for _, row in df.iterrows():
table_relevance += (
f'Table: {row["table_name"]}, relevance score: {row["similarities"]}\n'
f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'
)
if len(most_similar_tables) > 0:
for table in most_similar_tables:
table_relevance += (
f"Table: {table}, relevance score: {max(df['similarities'])}\n"
f"Table: `{table}`, relevance score: {max(df['similarities'])}\n"
)
return table_relevance

Expand Down

0 comments on commit c97f66f

Please sign in to comment.