diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index f9785d6c..24070dc1 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -278,8 +278,6 @@ RUN pip install mpld3 && \ pip install pyldavis==3.2.2 && \ pip install mlxtend && \ pip install altair && \ - # b/183944405 pystan 3.x is not compatible with fbprophet. - pip install pystan==2.19.1.1 && \ pip install ImageHash && \ pip install ecos && \ pip install CVXcanon && \ @@ -302,7 +300,7 @@ RUN pip install mpld3 && \ pip install pyexcel-ods && \ pip install sklearn-pandas && \ pip install stemming && \ - pip install fbprophet && \ + pip install prophet && \ pip install holoviews && \ pip install geoviews && \ pip install hypertools && \ diff --git a/tests/test_fbprophet.py b/tests/test_fbprophet.py index ecb69697..60566f8b 100644 --- a/tests/test_fbprophet.py +++ b/tests/test_fbprophet.py @@ -3,7 +3,7 @@ import numpy as np import pandas as pd -from fbprophet import Prophet +from prophet import Prophet class TestFbProphet(unittest.TestCase): def test_fit(self):