colname in columns check (get_category_counts)

src/Dataset.py

@@ -270,6 +270,10 @@ class Dataset:
         if type(colname) != str:
             logging.error("parameter `colname` is not a string")
             raise ValueError(f"{colname} is not a string")
+        if colname not in self.dataframe.columns:
+            logging.error("column requested not in dataframe")
+            raise KeyError(f"{colname} is not a column in dataframe")
+
         if not (ascending is None or type(ascending) is bool):
             logging.error("parameter `ascending` is not a bool or None")
             raise ValueError(f"{ascending} is not a bool or None")