Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

remove warning for missing dependencies for main darts package #1986

Merged
merged 2 commits into from
Sep 14, 2023
Merged

remove warning for missing dependencies for main darts package #1986

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2ffa946
remove warning for missing dependencies for main darts package
dennisbader Sep 7, 2023
9d9ea39
Merge branch 'master' into feat/remove_missing_deps_warning
dennisbader Sep 14, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 3 additions & 5 deletions darts/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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. "
Expand All @@ -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
Expand Down