Dealing with vectorised concentrations in exposure model

This commit is contained in:
Nicolas Mounet 2021-04-28 14:20:13 +02:00
parent 5afa228fb5
commit 487c930f8f

View file

@ -701,7 +701,7 @@ class ExposureModel:
def integrate(fn, start, stop):
values = np.linspace(start, stop)
return np.trapz([fn(v) for v in values], values)
return np.trapz([fn(v) for v in values], values, axis=0)
for start, stop in self.exposed.presence.boundaries():
exposure += integrate(self.concentration_model.concentration, start, stop)