Skip to content

Commit c441378

Browse files
committed
0.6.16 a154
1 parent 570053b commit c441378

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_autots.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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())

0 commit comments

Comments
 (0)