Merge pull request #268 from jdnklau/master

Fix exponential gradient reduction without protected attribute (#267)
Trusted-AI · Sep 7, 2021 · 2c09e2f · 2c09e2f
2 parents fcda24e + 0b7f395
commit 2c09e2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aif360/sklearn/inprocessing/exponentiated_gradient_reduction.py b/aif360/sklearn/inprocessing/exponentiated_gradient_reduction.py
@@ -144,6 +144,6 @@ def predict_proba(self, X):
             ``self.classes_``.
         """
         if self.drop_prot_attr:
-            X = X.drop(self.prot_attr)
+            X = X.drop(self.prot_attr, axis=1)
 
         return self.model._pmf_predict(X)