Merge pull request #75 from rbroc/fix-pos

Fix pos_stats extraction

textdescriptives/about.py

@@ -1,3 +1,3 @@
 __title__ = "textdescriptives"
-__version__ = "1.1.0"  # the ONLY source of version ID
+__version__ = "1.1.1"  # the ONLY source of version ID
 __download_url__ = "https://github.com/HLasse/textdescriptives"

textdescriptives/dataframe_extract.py

@@ -60,13 +60,10 @@ def __init__(
 
         if "all" in metrics:
             for component in valid_metrics - set(["all"]):
-                if doc.has_extension(component) or component == "descriptive_stats":
-                    extraction.append(self.__unpack_extension(doc, component))
-
+                extraction.append(self.__unpack_extension(doc, component))
         else:
             for component in metrics:
-                if doc.has_extension(component) or component == "descriptive_stats":
-                    extraction.append(self.__unpack_extension(doc, component))
+                extraction.append(self.__unpack_extension(doc, component))
 
         if self.as_dict:
             self.out = reduce(lambda a, b: {**a, **b}, extraction)
@@ -96,6 +93,8 @@ def __unpack_extension(self, doc: Doc, extension: str) -> pd.DataFrame:
         # we only need the getter
         if extension == "descriptive_stats":
             values = self.__get_descriptive_stats_dict(doc)
+        elif extension == "pos_stats":
+            values = doc.get_extension('pos_proportions')[2](doc)
         else:
             values = doc.get_extension(extension)[2](doc)