change the father class of WeightedErrorRate to ClassificationMoment

Signed-off-by: Acorn <wanglj17@mails.tsinghua.edu.cn>
fairlearn · Aug 20, 2020 · acbdedc · acbdedc
1 parent 34fc5d1
commit acbdedc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/fairlearn/reductions/_moments/cdf_demographic_parity_moment.py b/fairlearn/reductions/_moments/cdf_demographic_parity_moment.py
@@ -36,7 +36,7 @@ def __init__(self, loss, y_range, difference_bound=None, grids=[], grid_num=41):
         super().__init__()
         self.utility_parity = UtilityParity(difference_bound=difference_bound)
         self.loss = loss
-        self.objective = WeightedErrorRate(self.loss)
+        self.objective = WeightedErrorRate()
         self.grids = grids
         self.grid_num = grid_num
         self.y_range = y_range

diff --git a/fairlearn/reductions/_moments/weighted_error_rate.py b/fairlearn/reductions/_moments/weighted_error_rate.py
@@ -18,8 +18,8 @@ class WeightedErrorRate(ClassificationMoment):
 
     short_name = "Weighted Error Rate"
 
-    def __init__(self, loss):
-        super(WeightedErrorRate, self).__init__(loss)
+    def __init__(self):
+        super(WeightedErrorRate, self).__init__()
 
     '''for what we need here is augmented data. Hence to avoid unnecessary calculation, we use
     augmented data having been calculated in regression_moment here and directly return in