From a6ceb5d6fba22ae2256c9e07d19c3534b02b52b8 Mon Sep 17 00:00:00 2001 From: Dennis Bader Date: Thu, 14 Sep 2023 10:56:39 +0200 Subject: [PATCH] remove warning for missing dependencies for main darts package (#1986) --- darts/models/__init__.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/darts/models/__init__.py b/darts/models/__init__.py index 08513bba7b..0dac1a280d 100644 --- a/darts/models/__init__.py +++ b/darts/models/__init__.py @@ -39,8 +39,6 @@ from darts.models.forecasting.tft_model import TFTModel from darts.models.forecasting.tide_model import TiDEModel from darts.models.forecasting.transformer_model import TransformerModel - - except ModuleNotFoundError: logger.warning( "Support for Torch based models not available. " @@ -51,17 +49,17 @@ try: from darts.models.forecasting.lgbm import LightGBMModel except ModuleNotFoundError: - LightGBMModel = NotImportedModule(module_name="LightGBM") + LightGBMModel = NotImportedModule(module_name="LightGBM", warn=False) try: from darts.models.forecasting.prophet_model import Prophet except ImportError: - Prophet = NotImportedModule(module_name="Prophet") + Prophet = NotImportedModule(module_name="Prophet", warn=False) try: from darts.models.forecasting.catboost_model import CatBoostModel except ModuleNotFoundError: - CatBoostModel = NotImportedModule(module_name="CatBoost") + CatBoostModel = NotImportedModule(module_name="CatBoost", warn=False) try: from darts.models.forecasting.croston import Croston