diff --git a/functions.py b/functions.py index b39502a..2652c8a 100644 --- a/functions.py +++ b/functions.py @@ -37,7 +37,7 @@ class Symbol: ) symbols.sort_values(by="Match", ascending=False, inplace=True) - if symbols["Match"].head().sum(): + if symbols["Match"].head().sum() < 300: symbols["Match"] = symbols.apply( lambda x: fuzz.partial_ratio( search.lower(), f"{x['Symbol']} {x['Issue_Name']}".lower() @@ -45,7 +45,7 @@ class Symbol: axis=1, ) symbols.sort_values(by="Match", ascending=False, inplace=True) - print(symbols.head()) + return list(zip(list(symbols["Symbol"]), list(symbols["Description"]))) def find_symbols(self, text: str):