fix missing imports

views_stepshifter/manager/stepshifter_manager.py

@@ -1,22 +1,21 @@
 from views_pipeline_core.managers.model_manager import ModelManager
 from views_pipeline_core.managers.path_manager import ModelPath
 from views_pipeline_core.models.outputs import generate_output_dict
-from views_pipeline_core.files.utils import read_log_file, create_log_file
+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
 from views_pipeline_core.evaluation.metrics import generate_metric_dict
+from views_pipeline_core.configs.pipeline import PipelineConfig
 from views_stepshifter.models.stepshifter import StepshifterModel
 from views_stepshifter.models.hurdle_model import HurdleModel
 # from views_forecasts.extensions import *
 import logging
+import time
+import pickle
+from datetime import datetime
 import pandas as pd
 import numpy as np
-import time
 import wandb
-from datetime import datetime
 from sklearn.metrics import mean_squared_error
-from views_pipeline_core.files.utils import read_dataframe
-from views_pipeline_core.configs.pipeline import PipelineConfig
-import pickle
 
 logger = logging.getLogger(__name__)
 

views_stepshifter/models/hurdle_model.py

@@ -3,6 +3,7 @@
 from views_stepshifter.models.validation import views_validate
 from sklearn.utils.validation import check_is_fitted
 import pandas as pd
+from typing import List, Dict
 
 
 class HurdleModel(StepshifterModel):