From 1585eec6fdd73a041daef0e76d8932cea37e047f Mon Sep 17 00:00:00 2001 From: markus Date: Mon, 25 Jan 2021 14:13:34 +0100 Subject: [PATCH] add type hinting to _generate_viral_loads --- cara/montecarlo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cara/montecarlo.py b/cara/montecarlo.py index 62229843..60ed82ae 100644 --- a/cara/montecarlo.py +++ b/cara/montecarlo.py @@ -65,7 +65,7 @@ class MCInfectedPopulation(models.Population): viral_load: typing.Optional[float] = None - def _generate_viral_loads(self): + def _generate_viral_loads(self) -> np.ndarray: kde_model = KernelDensity(kernel='gaussian', bandwidth=0.1) kde_model.fit(np.asarray(log_viral_load_frequencies)[0, :][:, np.newaxis], sample_weight=np.asarray(log_viral_load_frequencies)[1, :])