diff --git a/cara/apps/calculator/model_generator.py b/cara/apps/calculator/model_generator.py index e09667e3..a945bd72 100644 --- a/cara/apps/calculator/model_generator.py +++ b/cara/apps/calculator/model_generator.py @@ -188,9 +188,9 @@ def model_from_form(form: FormData, tmp_raw_form_data) -> models.Model: # of the infected and exposed occupants respectively. # I.e. (infected_activity, infected_expiration), (exposed_activity, exposed_expiration) - activity_dict = {'Office/Meeting': (('Seated', 'Talking'), ('Seated', 'Talking')), - 'Training': (('Light exercise', 'Talking'), ('Seated', 'Whispering')), - 'Workshop': (('Light exercise', 'Talking'), ('Light exercise', 'Talking'))} + activity_dict = {'office': (('Seated', 'Talking'), ('Seated', 'Talking')), + 'training': (('Light exercise', 'Talking'), ('Seated', 'Whispering')), + 'workshop': (('Light exercise', 'Talking'), ('Light exercise', 'Talking'))} (infected_activity, infected_expiration), (exposed_activity, exposed_expiration) = activity_dict[form.activity_type] # Converts these strings to Activity and Expiration instances