allowing default forecastables

grid-parity-exchange · Sep 8, 2021 · b498ff3 · b498ff3
1 parent b9695fb
commit b498ff3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/prescient/engine/forecast_helper.py b/prescient/engine/forecast_helper.py
@@ -112,10 +112,11 @@ def _get_forecastable_locations(model):
             yield (model.data['system'], 'reserve_requirement')
 
     for data, key in _get_forecastable_locations(model):
-        if (not key in data or \
+        if (key not in data or \
             type(data[key]) is not dict or \
             data[key]['data_type'] != 'time_series' or \
             len(data[key]['values'] != num_entries)
            ):
+            default = None if (key not in data) else data[key]
             data[key] = { 'data_type': 'time_series',
-                          'values': [None]*num_entries}
+                          'values': [default]*num_entries}