Use prophet (1.1) instead of fbprophet (0.7)

kats/detectors/prophet_detector.py

@@ -14,8 +14,8 @@
 
 import numpy as np
 import pandas as pd
-from fbprophet import Prophet
-from fbprophet.serialize import model_from_json, model_to_json
+from prophet import Prophet
+from prophet.serialize import model_from_json, model_to_json
 from kats.consts import DEFAULT_VALUE_NAME, TimeSeriesData
 from kats.detectors.detector import DetectorModel
 from kats.detectors.detector_consts import AnomalyResponse, ConfidenceBand

kats/detectors/threshold_detector.py

@@ -7,8 +7,8 @@
 
 import numpy as np
 import pandas as pd
-from fbprophet import Prophet
-from fbprophet.serialize import model_from_json, model_to_json
+from prophet import Prophet
+from prophet.serialize import model_from_json, model_to_json
 from kats.consts import DEFAULT_VALUE_NAME, TimeSeriesData
 from kats.detectors.detector import DetectorModel
 from kats.detectors.detector_consts import AnomalyResponse

kats/models/prophet.py

@@ -9,7 +9,7 @@
 import pandas as pd
 
 try:
-    from fbprophet import Prophet
+    from prophet import Prophet
 
     _no_prophet = False
 except ImportError:
@@ -131,7 +131,7 @@ def __init__(
         extra_regressors: Optional[List[Dict[str, Any]]] = None,
     ) -> None:
         if _no_prophet:
-            raise RuntimeError("requires fbprophet to be installed")
+            raise RuntimeError("requires prophet to be installed")
         super().__init__()
         self.growth = growth
         self.changepoints = changepoints
@@ -244,7 +244,7 @@ class ProphetModel(Model[ProphetParams]):
     def __init__(self, data: TimeSeriesData, params: ProphetParams) -> None:
         super().__init__(data, params)
         if _no_prophet:
-            raise RuntimeError("requires fbprophet to be installed")
+            raise RuntimeError("requires prophet to be installed")
         self.data: TimeSeriesData = data
         self._data_params_validation()
 

kats/tests/models/test_prophet_model.py

@@ -106,14 +106,14 @@ def setUpClass(cls) -> None:
 
         # pyre-fixme[2]: Parameter annotation cannot be `Any`.
         def mock_prophet_import(module: Any, *args: Any, **kwargs: Any) -> None:
-            if module == "fbprophet":
+            if module == "prophet":
                 raise ImportError
             else:
                 return original_import_fn(module, *args, **kwargs)
 
         cls.mock_imports = patch("builtins.__import__", side_effect=mock_prophet_import)
 
-    def test_fbprophet_not_installed(self) -> None:
+    def test_prophet_not_installed(self) -> None:
         # Unload prophet module so its imports can be mocked as necessary
         del sys.modules["kats.models.prophet"]
 

requirements.txt

@@ -5,8 +5,7 @@ matplotlib>=2.0.0
 numpy>=1.21,<1.22
 pandas>=1.0.4,<=1.3.5
 python-dateutil>=2.8.0
-pystan==2.19.1.1
-fbprophet==0.7.1
+prophet==1.1
 neuralprophet==0.3.2
 scikit-learn>=0.24.2
 scipy<1.8.0

test_requirements.txt

@@ -1,7 +1,6 @@
 LunarCalendar>=0.0.9
 ax-platform==0.2.4
-pystan==2.19.1.1
-fbprophet==0.7.1
+prophet==1.1
 gpytorch
 holidays>=0.10.2
 numba>=0.52.0