raising NotImplementedError for unknown ST policies

COVIDAnalytics · Mar 19, 2021 · 0874f4a · 0874f4a
1 parent 96ef907
commit 0874f4a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cost_functions/economic_cost/economic_data/economic_params.py b/cost_functions/economic_cost/economic_data/economic_params.py
@@ -1,16 +1,18 @@
 from collections import defaultdict
 
 UNEMPLOYMENT_COST = {"GM": 77510,"US": 100000}
+def not_implemented():
+    raise NotImplementedError
 
 # employment_impact in percentages
 EMPLOYMENT_IMPACT = {}
-EMPLOYMENT_IMPACT["GM"] = defaultdict(lambda:0, { "Lockdown": 2,
+EMPLOYMENT_IMPACT["GM"] = defaultdict(not_implemented, { "Lockdown": 2,
         "Social Distancing": 1,
         "No_Measure": 0})
 
 GDP_IMPACT = {}
 
-GDP_IMPACT["GM"] = defaultdict(lambda:0, { "Lockdown": -10,
+GDP_IMPACT["GM"] = defaultdict(not_implemented, { "Lockdown": -10,
                                            "Social Distancing": -5,
                                            "No_Measure": 0})