Merge pull request Trusted-AI#103 from IBM/nrkarthikeyan-patch-1

nrkarthikeyan · web-flow · commit 2e13da8cd00d · 2019-09-06T13:26:17.000-04:00
Update opt_tools.py to check for optimality properly
diff --git a/aif360/algorithms/preprocessing/optim_preproc_helpers/opt_tools.py b/aif360/algorithms/preprocessing/optim_preproc_helpers/opt_tools.py
@@ -380,7 +380,7 @@ def optimize(self, epsilon=1., dlist=[], verbose=True):
         prob = Problem(obj, constraints)
         prob.solve(verbose=verbose)
 
-        if prob.status == "optimal":
+        if prob.status in ["optimal", "optimal_inaccurate"]:
             print("Optimized Preprocessing: Objective converged to %f"
                 % (prob.value))
         else:
