fixed bug with expected new cases when short range interactions are defined

This commit is contained in:
Luis Aleixo 2024-04-20 13:50:10 +01:00
parent 585dba57e9
commit 00b5dc535d

View file

@ -1812,6 +1812,12 @@ class ExposureModel:
isinstance(self.exposed.number, IntPiecewiseConstant)):
raise NotImplementedError("Cannot compute expected new cases "
"with dynamic occupancy")
"""
The expect_new_cases should always take the long-range infection_probability and multiply by the occupants exposed to long-range.
"""
if self.short_range != ():
return nested_replace(self, {'short_range': ()}).infection_probability() * self.exposed.number / 100
return self.infection_probability() * self.exposed.number / 100