File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -884,7 +884,7 @@ def test_sklearn(self):
884884 from autots import create_regressor
885885 from autots .models .sklearn import MultivariateRegression , DatepartRegression , WindowRegression
886886
887- df = load_daily (long = False )
887+ df = load_daily (long = False ). bfill (). ffill ()
888888 forecast_length = 8
889889 df_train = df .iloc [:- forecast_length ]
890890 df_test = df .iloc [- forecast_length :]
@@ -968,7 +968,7 @@ def test_sklearn(self):
968968 n_jobs = n_jobs ,
969969 ** params
970970 )
971- model .fit (df_train . ffill () )
971+ model .fit (df_train )
972972 first_forecast = model .predict (future_regressor = future_regressor_forecast )
973973 # first_forecast.plot_grid(df)
974974 self .assertListEqual (first_forecast .forecast .index .tolist (), df_test .index .tolist ())
You can’t perform that action at this time.
0 commit comments