fix wordcloud

paulzierep · Aug 12, 2024 · c36530e · c36530e
1 parent dbeb0c9
commit c36530e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bin/create_wordcloud.py b/bin/create_wordcloud.py
@@ -10,7 +10,7 @@
 from wordcloud import WordCloud
 
 
-def prepare_data(table_path: str, name_col: str, stat_col: str) -> Dict:
+def prepare_data(table_path: str, name_col: str, stat_col: str) -> pd.Series:
     """
     Prepare data to create dictionary with key being the name and
     value the stat/counts