don't correct bug that wouldn't be backwards compatible

pph-collective · May 11, 2023 · 996fc74 · 996fc74
1 parent 4001bfa
commit 996fc74
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/integration_test.py b/tests/integration_test.py
@@ -427,10 +427,10 @@ def test_dissolution(params_integration, tmpdir):
         model.params.demographics.black.sex_type.MSM.drug_type.Inj.num_partners
     )
     for agent in model.pop.all_agents:
-        agent.mean_num_partners["Inj"] = 0
-        agent.mean_num_partners["Sex"] = 0
-        agent.mean_num_partners["SexInj"] = 0
-        agent.mean_num_partners["Social"] = 0
+        agent.mean_num_partners["Inj"] = -1
+        agent.mean_num_partners["Sex"] = -1
+        agent.mean_num_partners["SexInj"] = -1
+        agent.mean_num_partners["Social"] = -1
     model.pop.update_partner_targets()
 
     model.step(tmpdir)

diff --git a/titan/population.py b/titan/population.py
@@ -445,7 +445,7 @@ def update_partnerability(self, a):
         """
         for bond in self.params.classes.bond_types.keys():
             if a in self.partnerable_agents[bond]:
-                if len(a.partners[bond]) >= (
+                if len(a.partners[bond]) > (
                     a.target_partners[bond] * self.params.calibration.partnership.buffer
                 ):
                     self.partnerable_agents[bond].remove(a)