fix: added logging options to avoid pystan info

broomva · Jul 28, 2023 · a9345a3 · a9345a3
1 parent 4495652
commit a9345a3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/hyperopt_prophet/model.py b/hyperopt_prophet/model.py
@@ -28,10 +28,10 @@
 )
 
 import logging
-
-logging.getLogger("prophet").setLevel(logging.WARNING)
-logging.getLogger("cmdstanpy").setLevel(logging.WARNING)
-
+logger = logging.getLogger('cmdstanpy')
+logger.addHandler(logging.NullHandler())
+logger.propagate = False
+logger.setLevel(logging.CRITICAL)
 
 class ForecastModel(ABC, mlflow.pyfunc.PythonModel):
     @abstractmethod

diff --git a/hyperopt_prophet/training.py b/hyperopt_prophet/training.py
@@ -14,9 +14,10 @@
 from .utils import get_plotly_forecast, plotly_fig2pil
 
 import logging
-
-logging.getLogger("prophet").setLevel(logging.WARNING)
-logging.getLogger("cmdstanpy").setLevel(logging.WARNING)
+logger = logging.getLogger('cmdstanpy')
+logger.addHandler(logging.NullHandler())
+logger.propagate = False
+logger.setLevel(logging.CRITICAL)
 
 
 warnings.filterwarnings("ignore")