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

add type hinting to _generate_viral_loads

Browse source
This commit is contained in:
markus 2021-01-25 14:13:34 +01:00
parent 8f3e279045
commit 1585eec6fd
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cara/montecarlo.py
View file

@ -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, :])