From 0b7f395287edf2c629d02704ab70de514010554d Mon Sep 17 00:00:00 2001 From: Jannik Dunkelau Date: Mon, 6 Sep 2021 10:03:32 +0200 Subject: [PATCH] Fix exponential gradient reduction without protected attribute (#267) --- aif360/sklearn/inprocessing/exponentiated_gradient_reduction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aif360/sklearn/inprocessing/exponentiated_gradient_reduction.py b/aif360/sklearn/inprocessing/exponentiated_gradient_reduction.py index 2375cb29..8e0fd366 100644 --- 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)