diff --git a/views_stepshifter/manager/stepshifter_manager.py b/views_stepshifter/manager/stepshifter_manager.py index cf962a6..af9a964 100644 --- a/views_stepshifter/manager/stepshifter_manager.py +++ b/views_stepshifter/manager/stepshifter_manager.py @@ -6,7 +6,7 @@ from views_pipeline_core.evaluation.metrics import generate_metric_dict from views_stepshifter.models.stepshifter import StepshifterModel from views_stepshifter.models.hurdle_model import HurdleModel -from views_forecasts.extensions import * +# from views_forecasts.extensions import * import logging import pandas as pd import numpy as np diff --git a/views_stepshifter/models/hurdle_model.py b/views_stepshifter/models/hurdle_model.py index 15fe99f..0736162 100644 --- a/views_stepshifter/models/hurdle_model.py +++ b/views_stepshifter/models/hurdle_model.py @@ -1,8 +1,8 @@ from views_pipeline_core.managers.model_manager import ModelManager from views_stepshifter.models.stepshifter import StepshifterModel from views_stepshifter.models.validation import views_validate -from views_forecasts.extensions import * from sklearn.utils.validation import check_is_fitted +import pandas as pd class HurdleModel(StepshifterModel): diff --git a/views_stepshifter/models/stepshifter.py b/views_stepshifter/models/stepshifter.py index 04fd6c7..101150c 100644 --- a/views_stepshifter/models/stepshifter.py +++ b/views_stepshifter/models/stepshifter.py @@ -1,10 +1,11 @@ import pickle import numpy as np +import pandas as pd import logging from darts import TimeSeries from sklearn.utils.validation import check_is_fitted from typing import List, Dict -from views_forecasts.extensions import * +# from views_forecasts.extensions import * from views_stepshifter.models.validation import views_validate from views_pipeline_core.managers.model_manager import ModelManager diff --git a/views_stepshifter/models/validation.py b/views_stepshifter/models/validation.py index 473e091..e80de8f 100644 --- a/views_stepshifter/models/validation.py +++ b/views_stepshifter/models/validation.py @@ -2,7 +2,6 @@ import numpy as np import pandas as pd import logging -from views_forecasts.extensions import * logger = logging.getLogger(__name__)