fix: add option to disable tqdm in minuit

ComPWA · Feb 24, 2021 · 48fc799 · 48fc799
1 parent 83fc9c1
commit 48fc799
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/tensorwaves/optimizer/minuit.py b/src/tensorwaves/optimizer/minuit.py
@@ -2,6 +2,7 @@
 
 """Minuit2 adapter to the `iminuit.Minuit` package."""
 
+import logging
 import time
 from datetime import datetime
 from typing import Dict, Iterable, Optional, Union
@@ -84,7 +85,9 @@ def optimize(  # pylint: disable=too-many-locals
         parameter_handler = ParameterFlattener(initial_parameters)
         flattened_parameters = parameter_handler.flatten(initial_parameters)
 
-        progress_bar = tqdm()
+        progress_bar = tqdm(
+            disable=logging.getLogger().level > logging.WARNING
+        )
         n_function_calls = 0
 
         def update_parameters(pars: list) -> None: