TLOmodel/src/scripts/hiv/program_simplification/analysis_hiv_program_simplification.py at 93e07b306f7d3330974e7a42c2ea77f7150bf965 · UCL/TLOmodel · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
"""This Scenario file runs the model under different assumptions for the HIV program simplification


check the batch configuration gets generated without error:
tlo scenario-run --draw-only src/scripts/hiv/program_simplification/analysis_hiv_program_simplification.py

Test the scenario starts running without problems:
tlo scenario-run src/scripts/hiv/program_simplification/analysis_hiv_program_simplification.py

or execute a single run:
tlo scenario-run src/scripts/hiv/program_simplification/analysis_hiv_program_simplification.py --draw 1 0

Run on the batch system using:
tlo batch-submit src/scripts/hiv/program_simplification/analysis_hiv_program_simplification.py

Display information about a job:
tlo batch-job tlo_q1_demo-123 --tasks

Download result files for a completed job:
tlo batch-download calibration_script-2022-04-12T190518Z

if running locally need to parse each folder
tlo parse-log /Users/tmangal/PycharmProjects/TLOmodel/outputs/hiv_program_simplification-2025-07-24T160218Z/0/0

"""

from pathlib import Path
from typing import Dict

from scripts.hiv.program_simplification.scenario_definitions import ScenarioDefinitions
from tlo import Date, logging
from tlo.methods.fullmodel import fullmodel
from tlo.scenario import BaseScenario


class HIV_Progam_Elements(BaseScenario):
    def __init__(self):
        super().__init__()
        self.seed = 10
        self.start_date = Date(2010, 1, 1)
        self.end_date = Date(2051, 1, 1)
        self.pop_size = 100_000
        self._scenarios = self._get_scenarios()
        self.number_of_draws = len(self._scenarios)
        self.runs_per_draw = 10

    def log_configuration(self):
        return {
            'filename': 'hiv_program_simplification',
            'directory': Path('./outputs'),
            'custom_levels': {
                '*': logging.WARNING,
                'tlo.methods.demography': logging.INFO,
                'tlo.methods.demography.detail': logging.WARNING,
                'tlo.methods.healthburden': logging.INFO,
                'tlo.methods.healthsystem': logging.WARNING,
                'tlo.methods.healthsystem.summary': logging.INFO,
                'tlo.methods.hiv': logging.INFO,
                'tlo.methods.tb': logging.INFO,
            }
        }

    def modules(self):
        return (
            fullmodel(use_simplified_births=True,
                      module_kwargs={"HealthSystem": {"equip_availability": 'all'}})
        )

    def draw_parameters(self, draw_number, rng):
        if draw_number < len(self._scenarios):
            return list(self._scenarios.values())[draw_number]

    def _get_scenarios(self) -> Dict[str, Dict]:
        """Return the Dict with values for the parameters that are changed, keyed by a name for the scenario."""

        scenario_definitions = ScenarioDefinitions()

        return {
            "Status Quo": scenario_definitions.status_quo(),

            "Reduce HIV testing": scenario_definitions.reduce_HIV_test(),

            "Remove Viral Load Testing": scenario_definitions.remove_VL(),

            "Target Viral Load Testing": scenario_definitions.target_VL(),

            "Replace Viral Load Testing": scenario_definitions.replace_VL_with_TDF(),

            "Remove PrEP for FSW": scenario_definitions.remove_prep_fsw(),

            "Remove PrEP for AGYW": scenario_definitions.remove_prep_agyw(),

            "Switch to injectable PrEP": scenario_definitions.switch_to_injectable_prep(),

            "Remove TPT for PLHIV": scenario_definitions.remove_TPT(),

            "Targeted TPT": scenario_definitions.target_TPT(),

            "Remove VMMC": scenario_definitions.remove_vmmc(),

            "Target All Elements": scenario_definitions.target_all(),

            "Reduce All Elements": scenario_definitions.remove_all(),

            "Program Scale-up": scenario_definitions.scaleup(),
        }


if __name__ == '__main__':
    from tlo.cli import scenario_run

    scenario_run([__file__])