Skip to content

Commit 66688d1

Browse files
fixing typo in function name similart_tables_based_on_few_shot_examples -> similar_tables_based_on_few_shot_examples (#473)
1 parent 6a3df95 commit 66688d1

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

Diff for: dataherald/sql_generator/dataherald_finetuning_agent.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def get_docs_embedding(
182182
def cosine_similarity(self, a: List[float], b: List[float]) -> float:
183183
return round(np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b)), 4)
184184

185-
def similart_tables_based_on_few_shot_examples(self, df: pd.DataFrame) -> List[str]:
185+
def similar_tables_based_on_few_shot_examples(self, df: pd.DataFrame) -> List[str]:
186186
most_similar_tables = set()
187187
if self.few_shot_examples is not None:
188188
for example in self.few_shot_examples:
@@ -224,7 +224,7 @@ def _run(
224224
)
225225
df = df.sort_values(by="similarities", ascending=True)
226226
df = df.tail(TOP_TABLES)
227-
most_similar_tables = self.similart_tables_based_on_few_shot_examples(df)
227+
most_similar_tables = self.similar_tables_based_on_few_shot_examples(df)
228228
table_relevance = ""
229229
for _, row in df.iterrows():
230230
table_relevance += f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'

Diff for: dataherald/sql_generator/dataherald_sqlagent.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ def get_docs_embedding(
246246
def cosine_similarity(self, a: List[float], b: List[float]) -> float:
247247
return round(np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b)), 4)
248248

249-
def similart_tables_based_on_few_shot_examples(self, df: pd.DataFrame) -> List[str]:
249+
def similar_tables_based_on_few_shot_examples(self, df: pd.DataFrame) -> List[str]:
250250
most_similar_tables = set()
251251
if self.few_shot_examples is not None:
252252
for example in self.few_shot_examples:
@@ -288,7 +288,7 @@ def _run(
288288
)
289289
df = df.sort_values(by="similarities", ascending=True)
290290
df = df.tail(TOP_TABLES)
291-
most_similar_tables = self.similart_tables_based_on_few_shot_examples(df)
291+
most_similar_tables = self.similar_tables_based_on_few_shot_examples(df)
292292
table_relevance = ""
293293
for _, row in df.iterrows():
294294
table_relevance += f'Table: `{row["table_name"]}`, relevance score: {row["similarities"]}\n'

0 commit comments

Comments
 (0)