diff --git a/caimira/src/caimira/api/controller/virus_report_controller.py b/caimira/src/caimira/api/controller/virus_report_controller.py index 01da4ca4..19cbcb05 100644 --- a/caimira/src/caimira/api/controller/virus_report_controller.py +++ b/caimira/src/caimira/api/controller/virus_report_controller.py @@ -29,9 +29,11 @@ def submit_virus_form(form_data: typing.Dict, report_generation_parallelism: typ form_obj: VirusFormData = generate_form_obj(form_data=form_data, data_registry=data_registry) report_data: typing.Dict = generate_report(form_obj=form_obj, report_generation_parallelism=report_generation_parallelism) - + # Handle model representation - if report_data['model']: report_data['model'] = repr(report_data['model']) - for single_group_output in report_data['groups'].values(): del single_group_output['model'] # Model representation per group not needed + if report_data['model']: + report_data['model'] = repr(report_data['model']) + for single_group_output in report_data['groups'].values(): + del single_group_output['model'] # Model representation per group not needed return report_data