Fixed Parsing Error and added Info print

cmu-db · Jul 15, 2024 · 8b9c554 · 8b9c554
1 parent 5988799
commit 8b9c554
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/scripts/anonymization/src/configuration/config_parser.py b/scripts/anonymization/src/configuration/config_parser.py
@@ -53,9 +53,9 @@ def get_config(self):
 
         if dp_info is not None:
 
-            cat = self.__get_column_type("ignore", dp_info)
+            cat = self.__get_column_type("categorical", dp_info)
             ordi = self.__get_column_type("ordinal", dp_info)
-            ignore = self.__get_column_type("continuous", dp_info)
+            ignore = self.__get_column_type("ignore", dp_info)
 
             eps = dp_info.get("epsilon", "1.0")
             pre_eps = dp_info.get("pre_epsilon", "0.5")
@@ -77,6 +77,12 @@ def get_config(self):
                             )
                         )
 
+            print("Parsed the following columns:\n"
+                  + f"Categorical: {cat}\n"
+                  + f"Continuous: {cont}\n"
+                  + f"Ordinal: {ordi}\n"
+                  + f"Ignored: {ignore}")
+
             column_classes = DPColumnConfig(ignore,cat,cont,ordi)
 
             anon_config = DPConfig(