fix: update remaining utils.poisson to distributions.poisson

pph-collective · Dec 21, 2021 · ad3e734 · ad3e734
1 parent c503777
commit ad3e734
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/titan/features/high_risk.py b/titan/features/high_risk.py
@@ -7,6 +7,7 @@
 from .. import agent
 from .. import population
 from .. import model
+from ..distributions import poisson
 
 
 class HighRisk(base_feature.BaseFeature):
@@ -170,7 +171,7 @@ def update_partner_numbers(self, pop: "population.Population", amount: int):
         """
         for bond in self.agent.location.params.high_risk.partnership_types:
             self.agent.mean_num_partners[bond] += amount  # could be negative
-            self.agent.target_partners[bond] = utils.poisson(
-                self.agent.mean_num_partners[bond], pop.np_random
+            self.agent.target_partners[bond] = poisson(
+                pop.np_random, self.agent.mean_num_partners[bond]
             )
             pop.update_partnerability(self.agent)
diff --git a/titan/interactions/injection.py b/titan/interactions/injection.py
@@ -39,7 +39,7 @@ def get_num_acts(cls, model: "model.TITAN", rel: "agent.Relationship") -> int:
             min(agent_params.num_acts, partner_params.num_acts)
             * model.calibration.injection.act
         )
-        share_acts = poisson(mean_num_acts, model.np_random)
+        share_acts = poisson(model.np_random, mean_num_acts)
 
         if share_acts < 1:
             return 0