diff --git a/R/tests/testthat/test_diagnostics.R b/R/tests/testthat/test_diagnostics.R index 6e2f631..0bf919d 100644 --- a/R/tests/testthat/test_diagnostics.R +++ b/R/tests/testthat/test_diagnostics.R @@ -282,7 +282,7 @@ test_that("copy", { m1 <- add_regressor(m1, 'binary_feature') m1 <- fit.prophet(m1, df) m2 <- prophet:::prophet_copy(m1, cutoff) - changepoints <- changepoints[changepoints <= cutoff] + changepoints <- changepoints[changepoints < cutoff] expect_equal(prophet:::set_date(changepoints), m2$changepoints) expect_true('custom' %in% names(m2$seasonalities)) expect_true('binary_feature' %in% names(m2$extra_regressors)) diff --git a/python/fbprophet/forecaster.py b/python/fbprophet/forecaster.py index 7bee303..dd0b33e 100644 --- a/python/fbprophet/forecaster.py +++ b/python/fbprophet/forecaster.py @@ -1106,7 +1106,7 @@ class Prophet(object): history = df[df['y'].notnull()].copy() if history.shape[0] < 2: raise ValueError('Dataframe has less than 2 non-NaN rows.') - self.history_dates = pd.to_datetime(df['ds'].unique()).sort_values() + self.history_dates = pd.to_datetime(pd.Series(df['ds'].unique(), name='ds')).sort_values() history = self.setup_dataframe(history, initialize_scales=True) self.history = history diff --git a/python/fbprophet/tests/test_diagnostics.py b/python/fbprophet/tests/test_diagnostics.py index cacf202..f515e4f 100644 --- a/python/fbprophet/tests/test_diagnostics.py +++ b/python/fbprophet/tests/test_diagnostics.py @@ -350,7 +350,7 @@ class TestDiagnostics(TestCase): m1.add_regressor('binary_feature') m1.fit(df) m2 = diagnostics.prophet_copy(m1, cutoff=cutoff) - changepoints = changepoints[changepoints <= cutoff] + changepoints = changepoints[changepoints < cutoff] self.assertTrue((changepoints == m2.changepoints).all()) self.assertTrue('custom' in m2.seasonalities) self.assertTrue('binary_feature' in m2.extra_regressors) diff --git a/python/fbprophet/tests/test_serialize.py b/python/fbprophet/tests/test_serialize.py index 841ecfa..e69fb8c 100644 --- a/python/fbprophet/tests/test_serialize.py +++ b/python/fbprophet/tests/test_serialize.py @@ -41,7 +41,7 @@ class TestSerialize(TestCase): # Make sure json doesn't get too large in the future self.assertTrue(len(model_str) < 200000) z = json.loads(model_str) - self.assertEqual(z['__fbprophet_version'], '0.6.1.dev0') + self.assertEqual(z['__fbprophet_version'], '0.7') m2 = model_from_json(model_str)