From 4d3ff3c7428b107ab182c489d16551b990d8178a Mon Sep 17 00:00:00 2001 From: Phil Elson Date: Wed, 7 Jul 2021 12:43:39 +0200 Subject: [PATCH] Remove the repeats concept from the report generator (removed in 1d34db3a81e7fb99ce08dfaa1ca1efcd1766c0bf). --- cara/apps/calculator/__init__.py | 4 ++++ cara/apps/calculator/report_generator.py | 20 -------------------- 2 files changed, 4 insertions(+), 20 deletions(-) diff --git a/cara/apps/calculator/__init__.py b/cara/apps/calculator/__init__.py index f7a4834d..a6a2b4db 100644 --- a/cara/apps/calculator/__init__.py +++ b/cara/apps/calculator/__init__.py @@ -96,6 +96,7 @@ class ConcentrationModel(BaseRequestHandler): if self.settings.get("debug", False): from pprint import pprint pprint(requested_model_config) + start = datetime.datetime.now() try: form = model_generator.FormData.from_dict(requested_model_config) @@ -114,6 +115,9 @@ class ConcentrationModel(BaseRequestHandler): report_generator.build_report, base_url, form, ) report: str = await asyncio.wrap_future(report_task) + if self.settings.get("debug", False): + dt = (datetime.datetime.now() - start) + print(f'Report response time {dt.seconds}.{dt.microseconds}s') self.finish(report) diff --git a/cara/apps/calculator/report_generator.py b/cara/apps/calculator/report_generator.py index ab206ec9..91e9e147 100644 --- a/cara/apps/calculator/report_generator.py +++ b/cara/apps/calculator/report_generator.py @@ -18,13 +18,6 @@ from .model_generator import FormData from ... import dataclass_utils -@dataclasses.dataclass(frozen=True) -class RepeatEvents: - repeats: int - probability_of_infection: float - expected_new_cases: float - - def model_start_end(model: models.ExposureModel): t_start = min(model.exposed.presence.boundaries()[0][0], model.concentration_model.infected.presence.boundaries()[0][0]) @@ -46,18 +39,6 @@ def calculate_report_data(model: models.ExposureModel): exposed_occupants = model.exposed.number expected_new_cases = np.mean(model.expected_new_cases()) - repeated_events = [] - for n in [1, 2, 3, 4, 5]: - - repeat_model = dataclass_utils.replace(model, repeats=n) - repeated_events.append( - RepeatEvents( - repeats=n, - probability_of_infection=np.mean(repeat_model.infection_probability()), - expected_new_cases=np.mean(repeat_model.expected_new_cases()), - ) - ) - return { "times": times, "concentrations": concentrations, @@ -67,7 +48,6 @@ def calculate_report_data(model: models.ExposureModel): "exposed_occupants": exposed_occupants, "expected_new_cases": expected_new_cases, "scenario_plot_src": img2base64(_figure2bytes(plot(times, concentrations, model))), - "repeated_events": repeated_events, }