remove warning for missing dependencies for main darts package (#1986)

unit8co · Sep 14, 2023 · a6ceb5d · a6ceb5d
1 parent 59937f3
commit a6ceb5d
Showing 1 changed file with 3 additions and 5 deletions.
diff --git 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