From 55e1a1a8321ce54b265ac59732ee4fdfe3b6620d Mon Sep 17 00:00:00 2001 From: markus Date: Wed, 11 Nov 2020 12:01:28 +0100 Subject: [PATCH] fix ExposureModel attribute bugs --- cara/apps/expert.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cara/apps/expert.py b/cara/apps/expert.py index d87ad3f3..3a5f0326 100644 --- a/cara/apps/expert.py +++ b/cara/apps/expert.py @@ -403,13 +403,13 @@ class CARAStateBuilder(state.StateBuilder): class ExpertApplication: def __init__(self): default_scenario = state.DataclassInstanceState( - models.Model, + models.ExposureModel, state_builder=CARAStateBuilder(), ) default_scenario.dcs_update_from(baseline_model) # For the time-being, we have to initialise the select states. Careful # as values might not correspond to what the baseline model says. - default_scenario.infected.mask.dcs_select('No mask') + default_scenario.concentration_model.infected.mask.dcs_select('No mask') self.scenarios = (default_scenario,) self.scenario_names = ('Scenario 1',) self.views = (WidgetView(default_scenario),) @@ -449,7 +449,7 @@ class ExpertApplication: def on_duplicate_click(b): self.scenario_names += (self.scenario_names[tab_index] + " (copy)",) new_scenario = state.DataclassInstanceState( - models.Model, + models.ExposureModel, state_builder=CARAStateBuilder(), ) new_scenario.dcs_update_from(self.scenarios[tab_index].dcs_instance())