From a9e07f77f61ebb7e0d6f318a04e2c9e1882deda1 Mon Sep 17 00:00:00 2001 From: Nicolas Mounet Date: Thu, 12 Nov 2020 18:50:14 +0100 Subject: [PATCH] Extending the time range for the plot in the report_generator --- cara/apps/calculator/report_generator.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cara/apps/calculator/report_generator.py b/cara/apps/calculator/report_generator.py index b78d5069..6a9f4648 100644 --- a/cara/apps/calculator/report_generator.py +++ b/cara/apps/calculator/report_generator.py @@ -24,8 +24,10 @@ class RepeatEvents: def calculate_report_data(model: models.ExposureModel): resolution = 600 - t_start = model.exposed.presence.boundaries()[0][0] - t_end = model.exposed.presence.boundaries()[-1][1] + t_start = min(model.exposed.presence.boundaries()[0][0], + model.concentration_model.infected.presence.boundaries()[0][0]) + t_end = max(model.exposed.presence.boundaries()[-1][1], + model.concentration_model.infected.presence.boundaries()[-1][1]) times = list(np.linspace(t_start, t_end, resolution)) concentrations = [model.concentration_model.concentration(time) for time in times]