From b76ace9570a6328aa6ec9156329695933c45b9e5 Mon Sep 17 00:00:00 2001 From: Ben Letham Date: Mon, 3 Dec 2018 15:43:13 -0800 Subject: [PATCH] Test type casting of extra regressors --- python/fbprophet/tests/test_prophet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/fbprophet/tests/test_prophet.py b/python/fbprophet/tests/test_prophet.py index 9686548..f807c85 100644 --- a/python/fbprophet/tests/test_prophet.py +++ b/python/fbprophet/tests/test_prophet.py @@ -589,7 +589,7 @@ class TestProphet(TestCase): ) m.add_regressor('binary_feature2', standardize=True) df = DATA.copy() - df['binary_feature'] = [0] * 255 + [1] * 255 + df['binary_feature'] = ['0'] * 255 + ['1'] * 255 df['numeric_feature'] = range(510) df['numeric_feature2'] = range(510) with self.assertRaises(ValueError):