pt
/
superset
mirror of https://github.com/apache/superset.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore(prophet): ignore logged warning for missing plotly package (#11334) 

* chore(prophet): ignore logged warning for missing plotly package

* reset log level after import
This commit is contained in:
Ville Brofeldt 2020-10-22 07:27:02 +03:00 committed by GitHub
parent a2a614d760
commit dcf12762b3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
superset/utils/pandas_postprocessing.py
View File

@ -14,6 +14,7 @@
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
import logging
from functools import partial
from typing import Any, Callable, cast, Dict, List, Optional, Tuple, Union
@ -590,7 +591,12 @@ def _prophet_fit_and_predict( # pylint: disable=too-many-arguments
Fit a prophet model and return a DataFrame with predicted results.
"""
try:
prophet_logger = logging.getLogger("fbprophet.plot")
prophet_logger.setLevel(logging.CRITICAL)
from fbprophet import Prophet # pylint: disable=import-error
prophet_logger.setLevel(logging.NOTSET)
except ModuleNotFoundError:
raise QueryObjectValidationError(_("`fbprophet` package not installed"))
model = Prophet(