Updated Arima to use pmdarima rather than pyramid-arima (#212)

datasciencecampus · Mar 25, 2019 · f691d3f · f691d3f
1 parent 715737b
commit f691d3f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/scripts/algorithms/arima.py b/scripts/algorithms/arima.py
@@ -1,5 +1,5 @@
 from numpy import clip, inf
-from pyramid.arima import auto_arima
+from pmdarima.arima import auto_arima
 
 
 class ARIMAForecast(object):

diff --git a/setup.py b/setup.py
@@ -54,7 +54,7 @@ def setup_package():
             'Programming Language :: Python :: 3.6',
         ],
         install_requires=['matplotlib', 'numpy', 'scipy', 'wordcloud', 'pandas', 'tqdm', 'nltk', 'scikit-learn', 'xlrd',
-                          'python-Levenshtein', 'gensim', 'pyramid-arima>=0.9.0', 'keras', 'tensorflow', 'keras_tqdm',
+                          'python-Levenshtein', 'gensim', 'pmdarima', 'keras', 'tensorflow', 'keras_tqdm',
                           'patsy', 'humanfriendly', 'psutil', 'jinja2'],
         # extras_require={'dev': ['check-manifest'],'test': ['coverage'],},
         python_requires='>=3.6',

diff --git a/tests/algorithms/test_arima.py b/tests/algorithms/test_arima.py
@@ -39,4 +39,4 @@ def test_linearly_increasing_sequence(self):
 
         actual_prediction = arima.predict_counts()
 
-        np_test.assert_almost_equal(actual_prediction, expected_prediction, decimal=1)
+        np_test.assert_almost_equal(actual_prediction, expected_prediction, decimal=0)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -39,4 +39,4 @@ def test_linearly_increasing_sequence(self):

		actual_prediction = arima.predict_counts()

		np_test.assert_almost_equal(actual_prediction, expected_prediction, decimal=1)
		np_test.assert_almost_equal(actual_prediction, expected_prediction, decimal=0)