Reformatting (hopefully last round)

IBM · Jun 22, 2023 · e33134d · e33134d
1 parent 93505a8
commit e33134d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/lale/lib/aif360/_mystic_util.py b/lale/lib/aif360/_mystic_util.py
@@ -46,7 +46,7 @@ def obtain_solver_info(
             group_mapping[g] = group_mapping[g][:-1] + str(new)
 
     mapped_osizes = {k1: osizes[k2] for k1, k2 in group_mapping.items()}
-    
+
     # calculate di ratios and invert if needed
     odi = []
     num_prot_attr = len(list(group_mapping.keys())[0]) - 1
@@ -66,7 +66,7 @@ def obtain_solver_info(
         calc_di = (
             (disadv_grp_adv_cls_ct) / (disadv_grp_adv_cls_ct + disadv_grp_disadv_cls_ct)
         ) / ((adv_grp_adv_cls_ct) / (adv_grp_adv_cls_ct + adv_grp_disadv_cls_ct))
-        
+
         if calc_di <= 1:
             odi.append(calc_di)
         else:
@@ -82,7 +82,7 @@ def obtain_solver_info(
 
     # recompute mapping based on any flipping of protected attribute values
     mapped_osizes = {k1: osizes[k2] for k1, k2 in group_mapping.items()}
-    
+
     sorted_osizes = [x[1] for x in sorted(mapped_osizes.items(), key=lambda x: x[0])]
 
     # construct variables for solver

diff --git a/lale/lib/aif360/urbis.py b/lale/lib/aif360/urbis.py
@@ -54,10 +54,10 @@ def _pick_sizes(
     group_mapping, o_flat, nci_vec, ndi_vec = obtain_solver_info(
         osizes, imbalance_repair_level, bias_repair_level, favorable_labels
     )
-    
+
     # pass into solver
     n_flat = calc_undersample_soln(o_flat, favorable_labels, nci_vec, ndi_vec)
-    
+
     return parse_solver_soln(n_flat, group_mapping)
 
 

diff --git a/test/test_aif360.py b/test/test_aif360.py
@@ -664,7 +664,7 @@ def test_sans_mitigation_pd_num(self):
 class TestAIF360UrbisPickSizes(unittest.TestCase):
     def setUp(self):
         from mystic.tools import random_seed
-        
+
         random_seed(42)
 
     def test_urbis_pick_sizes_single_pa_single_class_normal(self):