Disable prophet model

forecastlab · Oct 5, 2023 · d5492fa · d5492fa
1 parent c558431
commit d5492fa
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/dash/pages/search.py b/dash/pages/search.py
@@ -391,11 +391,14 @@ def filter_results(**kwargs):
 
         for item_title in unique_series_titles[0:show_num]:
             try:
-                forecast_series_dicts[item_title] = get_forecast_data(
+                series_data = get_forecast_data(
                     item_title
                 )
+                if series_data == None:
+                    raise FileNotFoundError
+
+                forecast_series_dicts[item_title] = series_data
 
-                series_data = forecast_series_dicts[item_title]
                 url_title = urlencode({"title": item_title})
 
                 title = (

diff --git a/dash/pages/series.py b/dash/pages/series.py
@@ -112,6 +112,8 @@ def _forecast_performance_layout():
 def _get_series_title(title):
     try:
         series_data_dict = get_forecast_data(title)
+        if series_data_dict == None:
+            raise Exception
     except:
         return "Series"
 
@@ -361,6 +363,8 @@ def wrapper(*args):
             if "title" in parse_result:
                 title = parse_result["title"][0]
                 series_data_dict = get_forecast_data(title)
+                if series_data_dict == None:
+                    raise PreventUpdate
 
                 del kwargs_dict[location_id]
                 return func(series_data_dict, **kwargs_dict)

diff --git a/updater/run_models.py b/updater/run_models.py
@@ -37,7 +37,7 @@
     "RComb",
     "LinearRegressionForecast",
     "RNN_M4_benchmark",
-    "FBProphet",
+    # "FBProphet",
     "RCES",
 ]