From cdbd8fa00a619893ecfa31622d353a67dcea0d15 Mon Sep 17 00:00:00 2001
From: xiaolongsun <95378566+xiaolong0728@users.noreply.github.com>
Date: Wed, 11 Dec 2024 11:17:07 +0100
Subject: [PATCH] fix missing imports

---
 views_stepshifter/manager/stepshifter_manager.py | 11 +++++------
 views_stepshifter/models/hurdle_model.py         |  1 +
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/views_stepshifter/manager/stepshifter_manager.py b/views_stepshifter/manager/stepshifter_manager.py
index af9a964..45d1e8a 100644
--- a/views_stepshifter/manager/stepshifter_manager.py
+++ b/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__)
 
diff --git a/views_stepshifter/models/hurdle_model.py b/views_stepshifter/models/hurdle_model.py
index 0736162..ae8ff7b 100644
--- a/views_stepshifter/models/hurdle_model.py
+++ b/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):