Merge pull request #174 from dssg/fix-folktables-bug

Change attribute call to sensitive_column
dssg · Feb 20, 2024 · 3ca75c5 · 3ca75c5
2 parents 37bf697 + c8c80ff
commit 3ca75c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/aequitas/flow/datasets/folktables.py b/src/aequitas/flow/datasets/folktables.py
@@ -244,7 +244,7 @@ def load_data(self):
             self.data[col] = (
                 self.data[col]
                 .replace(2, 0)
-                .astype("category" if col == self.sensitive_feature else bool)
+                .astype("category" if col == self.sensitive_column else bool)
             )
 
         if self.sensitive_column == "AGEP_bin":