in-place operation (#1)

InstituteforDiseaseModeling · Nov 22, 2024 · a792730 · a792730
1 parent 673045f
commit a792730
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/laser_generic/transmission.py b/src/laser_generic/transmission.py
@@ -113,7 +113,8 @@ def __call__(self, model, tick) -> None:
 
         np.multiply(contagion, beta_effective, out=forces)
         np.divide(forces, model.patches.populations[tick, :], out=forces)  # per agent force of infection
-        np.expm1(-forces, out=forces)
+        np.negative(forces, out=forces)
+        np.expm1(forces, out=forces)
         np.negative(forces, out=forces)
 
         # TODO: This is a hack to handle the different transmission dynamics for all of these SIS, SI, SIR, SEIR, ... models.