From d2f75f22375d8ef9af80094e796a73c3e9b0cab1 Mon Sep 17 00:00:00 2001
From: Vincent Roseberry <rosbo@google.com>
Date: Mon, 25 Apr 2022 23:41:20 +0000
Subject: [PATCH] Upgrade to latest version of prophet (1.0.1)

- The package has changed names: https://github.com/facebook/prophet#version-10-20210328
---
 Dockerfile.tmpl         | 4 +---
 tests/test_fbprophet.py | 2 +-
 2 files changed, 2 insertions(+), 4 deletions(-)

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):