Skip to content

Update the correct column if HIV HSI event didn't run #1835

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tamuri merged 2 commits into from
Feb 26, 2026
Merged

Update the correct column if HIV HSI event didn't run #1835

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8bdbb8f
Update the correct column if event didn't run
tamuri Feb 25, 2026
72fc2fe
Put logic for prep column in its own method
tamuri Feb 25, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions src/tlo/methods/hiv.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3443,13 +3443,16 @@ def __init__(self, module, person_id, type_of_prep):
self.counter_for_drugs_not_available = 0
self.type_of_prep = type_of_prep

def get_prep_column(self):
return 'hv_is_on_prep_oral' if self.type_of_prep == 'oral' else 'hv_is_on_prep_inj'

def apply(self, person_id, squeeze_factor):
"""Start PrEP for this person; or continue them on PrEP for 3 more months"""

df = self.sim.population.props
p = self.module.parameters
person = df.loc[person_id]
property = 'hv_is_on_prep_oral' if self.type_of_prep == 'oral' else 'hv_is_on_prep_inj'
prep_column = self.get_prep_column()
days_on_prep_property = 'hv_days_on_oral_prep' if self.type_of_prep == 'oral' else 'hv_days_on_inj_prep'

# Do not run if the person is not alive or is diagnosed with hiv
Expand Down Expand Up @@ -3484,7 +3487,7 @@ def apply(self, person_id, squeeze_factor):
if self.get_consumables(
item_codes={self.module.item_codes_for_consumables_required['prep']: days_on_prep}
):
df.at[person_id, property] = True
df.at[person_id, prep_column] = True

if df.at[person_id, "li_is_sexworker"]:
df.at[person_id, f'{days_on_prep_property}_FSW'] += days_on_prep
Expand All @@ -3504,7 +3507,7 @@ def apply(self, person_id, squeeze_factor):

else:
# If PrEP is not available, the person will default and not be on PrEP
df.at[person_id, property] = False
df.at[person_id, prep_column] = False

self.counter_for_drugs_not_available += (
1 # The current appointment is included in the count.
Expand All @@ -3525,7 +3528,8 @@ def apply(self, person_id, squeeze_factor):
def never_ran(self):
"""This is called if this HSI was never run.
Default the person to being off PrEP"""
self.sim.population.props.at[self.target, property] = False
prep_column = self.get_prep_column()
self.sim.population.props.at[self.target, prep_column] = False


class HSI_Hiv_StartOrContinueTreatment(HSI_Event, IndividualScopeEventMixin):
Expand Down