From 2a2443f8fc5b9b99a4c7a8ed2c89e666b2c4b216 Mon Sep 17 00:00:00 2001 From: Dilawar Singh Date: Sat, 24 Jul 2021 10:10:37 +0530 Subject: [PATCH] few minor tweaks to keep mypy happy in plots.py file Related: - https://github.com/COVID-IWG/epimargin/issues/124 - https://github.com/COVID-IWG/epimargin/issues/123 --- epimargin/plots.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/epimargin/plots.py b/epimargin/plots.py index 7003b44..0f61bdc 100644 --- a/epimargin/plots.py +++ b/epimargin/plots.py @@ -1,7 +1,7 @@ import datetime from collections import namedtuple from pathlib import Path -from typing import Optional, Sequence, Tuple +from typing import Optional, Sequence, Tuple, List, Dict import matplotlib as mpl import matplotlib.dates as mdates @@ -391,11 +391,11 @@ def simulations( num_sims = len(simulation_results) total_time = len(policy_outcomes[0][0]) - ranges = [{"max": [], "min": [], "mdn": [], "avg": []} for _ in range(len(policy_outcomes))] + ranges : List[Dict[str, List]] = [{"max": [], "min": [], "mdn": [], "avg": []} for _ in range(len(policy_outcomes))] for (i, policy) in enumerate(policy_outcomes): - for t in range(total_time): - curve_sorted = sorted([curve[t] for curve in policy]) + for _t in range(total_time): + curve_sorted = sorted([curve[_t] for curve in policy]) ranges[i]["min"].append(curve_sorted[0]) ranges[i]["max"].append(curve_sorted[-1]) ranges[i]["mdn"].append(curve_sorted[num_sims//2])