diff --git a/kats/data/README b/kats/data/README index 79bbeb356..92123019a 100644 --- a/kats/data/README +++ b/kats/data/README @@ -4,15 +4,15 @@ Please note: it's always dangerous to upload data set to fbcode, so please keep the data set as small as possible, it's just for testing purpose. 1. air_passengers.csv - See more details from fbprophet github: https://github.com/facebook/prophet/tree/master/examples + See more details from prophet github: https://github.com/facebook/prophet/tree/master/examples 2. mean_change_detection_test.csv Simulated example time series to test mean change detection. The real change point for increase change is 30 and 50 for decrease. See n165038 for how to generated the time series. 3. cdn_working_set.csv 4. multi_ts.csv 5. peyton_manning.csv - See more details from fbprophet github: https://github.com/facebook/prophet/tree/master/examples + See more details from prophet github: https://github.com/facebook/prophet/tree/master/examples 6. retail_sales.csv - See more details from fbprophet github: https://github.com/facebook/prophet/tree/master/examples + See more details from prophet github: https://github.com/facebook/prophet/tree/master/examples 7. yosemite_temps.csv - See more details from fbprophet github: https://github.com/facebook/prophet/tree/master/examples + See more details from prophet github: https://github.com/facebook/prophet/tree/master/examples diff --git a/kats/detectors/prophet_detector.py b/kats/detectors/prophet_detector.py index f7f646b36..865cc6d7f 100644 --- a/kats/detectors/prophet_detector.py +++ b/kats/detectors/prophet_detector.py @@ -14,8 +14,8 @@ import pandas as pd try: - 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 _no_prophet = False except ImportError: @@ -144,7 +144,7 @@ def __init__( uncertainty_samples: float = 50, ) -> None: if _no_prophet: - raise RuntimeError("requires fbprophet to be installed") + raise RuntimeError("requires prophet to be installed") if serialized_model: self.model = model_from_json(serialized_model) else: diff --git a/kats/models/prophet.py b/kats/models/prophet.py index c840b9211..8ed5056fc 100644 --- a/kats/models/prophet.py +++ b/kats/models/prophet.py @@ -8,7 +8,7 @@ import pandas as pd try: - from fbprophet import Prophet + from prophet import Prophet _no_prophet = False except ImportError: @@ -102,7 +102,7 @@ def __init__( custom_seasonalities: Optional[List[Dict]] = 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 @@ -205,7 +205,7 @@ class ProphetModel(m.Model): 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") if not isinstance(self.data.value, pd.Series): msg = "Only support univariate time series, but get {type}.".format( type=type(self.data.value) diff --git a/requirements.txt b/requirements.txt index d7b325132..8ff733b09 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,4 +7,4 @@ python-dateutil>=2.8.0 scikit-learn>+0.24.2 seaborn>=0.11.1 setuptools-git>=1.2 -statsmodels>=0.12.2 +statsmodels>=0.12.2,<0.13.0 diff --git a/test_requirements.txt b/test_requirements.txt index 4f1c1be02..80f71e546 100644 --- a/test_requirements.txt +++ b/test_requirements.txt @@ -1,7 +1,7 @@ LunarCalendar>=0.0.9 botorch==0.3.2 ax-platform==0.1.18 -fbprophet==0.7 +prophet==1.0 neuralprophet>=0.2.7 gpytorch==1.2.1 holidays>=0.10.2