Skip to content

Commit

Permalink
remove debugging code
Browse files Browse the repository at this point in the history
  • Loading branch information
mmcleod89 committed May 15, 2024
1 parent 5260b13 commit 5780eb3
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions src/hivpy/sexual_behaviour.py
Original file line number Diff line number Diff line change
Expand Up @@ -317,15 +317,12 @@ def prob_transition(self, sex_class: SexBehaviourClass, risk, i, j):
transition_matrix = self.sex_behaviour_trans[sex_class]

denominator = transition_matrix[i][0] + risk*sum(transition_matrix[i][1:])
# warnings.filterwarnings("error")
# try:

if (j == 0):
Probability = transition_matrix[i][0] / denominator
else:
Probability = risk*transition_matrix[i][j] / denominator
# except RuntimeWarning:
# breakpoint()
# warnings.resetwarnings()

return Probability

def get_partners_for_group(self, sex_class, group, size):
Expand Down

0 comments on commit 5780eb3

Please sign in to comment.