Add description to progress bar

Signed-off-by: Fred Branchaud <frederic.branchaud-charron@elementai.com>
fairlearn · Aug 14, 2020 · 83cf67f · 83cf67f
1 parent 69ca9e1
commit 83cf67f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGES.md b/CHANGES.md
@@ -41,6 +41,7 @@
 * Add new constraints and objectives in `ThresholdOptimizer`
 * Add class `InterpolatedThresholder` to represent the fitted `ThresholdOptimizer`
 * Add `fairlearn.datasets` module.
+* Add optional `verbose` argument to `GridSearch.fit` to diplay progress using `tqdm`.
 
 ### v0.4.6
 

diff --git a/fairlearn/reductions/_grid_search/grid_search.py b/fairlearn/reductions/_grid_search/grid_search.py
@@ -157,7 +157,7 @@ def fit(self, X, y, **kwargs):
 
         # Fit the estimates
         logger.debug("Setup complete. Starting grid search")
-        grid_iter = tqdm(grid.columns) if verbose else grid.columns
+        grid_iter = tqdm(grid.columns, desc="Grid search progress") if verbose else grid.columns
         for i in grid_iter:
             lambda_vec = grid[i]
             logger.debug("Obtaining weights")