pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added mypy typing check

Browse source
This commit is contained in:
Luis Aleixo 2023-06-21 12:16:47 +02:00
parent b6051778eb
commit fe60e1703f
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
caimira/apps/calculator/report_generator.py
View file

@ -197,8 +197,8 @@ def uncertainties_plot(exposure_model: models.ExposureModel, prob: typing.Union[
min_vl, max_vl, step = 2, 10, 8/100.
viral_loads = np.arange(min_vl, max_vl, step)
pi_means, lower_percentiles, upper_percentiles = [], [], []
for vl_log_min in viral_loads:
specific_prob = infection_probability[np.where((vl_log_min-vl)*(vl_log_min+step-vl)<0)[0]]
for vl_log in viral_loads:
specific_prob = infection_probability[np.where((vl_log-vl)*(vl_log+step-vl)<0)[0]] #type: ignore
pi_means.append(specific_prob.mean())
lower_percentiles.append(np.quantile(specific_prob, 0.05))
upper_percentiles.append(np.quantile(specific_prob, 0.95))