Merge pull request #4 from views-platform/feature_001_refactor_managers

Change ModelPath to ModelPathManager

views_stepshifter/manager/stepshifter_manager.py

@@ -1,5 +1,4 @@
-from views_pipeline_core.managers.model_manager import ModelManager
-from views_pipeline_core.managers.path_manager import ModelPath
+from views_pipeline_core.managers.model import ModelPathManager, ModelManager
 from views_pipeline_core.models.outputs import generate_output_dict
 from views_pipeline_core.files.utils import read_log_file, create_log_file, read_dataframe
 from views_pipeline_core.wandb.utils import add_wandb_monthly_metrics, generate_wandb_log_dict, log_wandb_log_dict
@@ -22,7 +21,7 @@
 
 class StepshifterManager(ModelManager):
 
-    def __init__(self, model_path: ModelPath) -> None:
+    def __init__(self, model_path: ModelPathManager) -> None:
         super().__init__(model_path)
         self._is_hurdle = self._config_meta["algorithm"] == "HurdleModel"
 

views_stepshifter/models/hurdle_model.py

@@ -1,4 +1,4 @@
-from views_pipeline_core.managers.model_manager import ModelManager
+from views_pipeline_core.managers.model import ModelManager
 from views_stepshifter.models.stepshifter import StepshifterModel
 from views_stepshifter.models.validation import views_validate
 from sklearn.utils.validation import check_is_fitted

views_stepshifter/models/stepshifter.py

@@ -7,7 +7,7 @@
 from typing import List, Dict
 # from views_forecasts.extensions import *
 from views_stepshifter.models.validation import views_validate
-from views_pipeline_core.managers.model_manager import ModelManager
+from views_pipeline_core.managers.model import ModelManager
 
 logger = logging.getLogger(__name__)