Moved forecasting horizon out

Author: christopherbunn

evalml/pipelines/components/estimators/regressors/arima_regressor.py

@@ -5,6 +5,7 @@
 import pandas as pd
 from pandas.api.types import is_integer_dtype
 from skopt.space import Integer
+from sktime.forecasting.base import ForecastingHorizon
 
 from evalml.model_family import ModelFamily
 from evalml.pipelines.components.estimators import Estimator
@@ -141,8 +142,6 @@ def _remove_datetime(
         return data_no_dt
 
     def _set_forecast(self, X: pd.DataFrame):
-        from sktime.forecasting.base import ForecastingHorizon
-
         # we can only calculate the difference if the indices are of the same type
         units_diff = 1
         if isinstance(X.index[0], type(self.last_X_index)) and isinstance(

evalml/pipelines/components/estimators/regressors/varmax_regressor.py

@@ -4,6 +4,7 @@
 import numpy as np
 import pandas as pd
 from skopt.space import Categorical, Integer
+from sktime.forecasting.base import ForecastingHorizon
 
 from evalml.model_family import ModelFamily
 from evalml.pipelines.components.estimators import Estimator
@@ -92,8 +93,6 @@ def __init__(
         )
 
     def _set_forecast_horizon(self, X: pd.DataFrame):
-        from sktime.forecasting.base import ForecastingHorizon
-
         # we can only calculate the difference if the indices are of the same type
         units_diff = 1
         if isinstance(X.index[0], type(self.last_X_index)):