Skip to content

Commit eff34fb

Browse files
committed
Add function emoji_df to get the dataframe
1 parent 677f11f commit eff34fb

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

advertools/emoji.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@
100100
101101
"""
102102

103-
__all__ = ["extract_emoji", "emoji_search"]
103+
__all__ = ["extract_emoji", "emoji_search", "emoji_df"]
104104

105105
import re
106106
from collections import Counter
@@ -116,7 +116,8 @@
116116

117117
def _emoji_df_path() -> str:
118118
from os.path import join
119-
return join("advertools", "pkg_data", "emoji_df.parquet")
119+
120+
return join(adv.__path__[0], "pkg_data", "emoji_df.parquet")
120121

121122

122123
def _emoji_entries(emoji):
@@ -274,3 +275,8 @@ def emoji_search(regex):
274275
.apply(lambda row: any(row), axis=1)
275276
)
276277
return emoji_df[result_index].reset_index(drop=True)
278+
279+
280+
def emoji_df():
281+
df = pd.read_parquet(_emoji_df_path())
282+
return df

0 commit comments

Comments
 (0)