|
9 | 9 | from tlo.analysis.utils import parse_log_file
|
10 | 10 | from tlo.methods import (
|
11 | 11 | bladder_cancer,
|
| 12 | + care_of_women_during_pregnancy, |
| 13 | + contraception, |
12 | 14 | demography,
|
13 | 15 | enhanced_lifestyle,
|
14 | 16 | healthburden,
|
15 | 17 | healthseekingbehaviour,
|
16 | 18 | healthsystem,
|
17 |
| - schisto, |
18 |
| - simplified_births, |
| 19 | + labour, |
| 20 | + newborn_outcomes, |
| 21 | + postnatal_supervisor, |
| 22 | + pregnancy_supervisor, |
19 | 23 | symptommanager,
|
20 | 24 | )
|
21 | 25 |
|
|
34 | 38 |
|
35 | 39 | start_date = Date(2010, 1, 1)
|
36 | 40 | end_date = Date(2013, 1, 1)
|
37 |
| -popsize = 5000 |
| 41 | +popsize = 19000 |
38 | 42 |
|
39 | 43 | # Establish the simulation object
|
40 | 44 | log_config = {
|
41 | 45 | 'filename': 'LogFile',
|
42 | 46 | 'directory': outputpath,
|
43 | 47 | 'custom_levels': {
|
44 |
| - "*": logging.WARNING, |
45 |
| - 'tlo.methods.bladder_cancer': logging.INFO, |
46 |
| - 'tlo.methods.schisto': logging.INFO, |
| 48 | + 'tlo.methods.demography': logging.CRITICAL, |
| 49 | + 'tlo.methods.contraception': logging.CRITICAL, |
| 50 | + 'tlo.methods.healthsystem': logging.CRITICAL, |
| 51 | + 'tlo.methods.labour': logging.CRITICAL, |
| 52 | + 'tlo.methods.healthburden': logging.CRITICAL, |
| 53 | + 'tlo.methods.symptommanager': logging.CRITICAL, |
| 54 | + 'tlo.methods.healthseekingbehaviour': logging.CRITICAL, |
| 55 | + 'tlo.methods.pregnancy_supervisor': logging.CRITICAL |
47 | 56 | }
|
48 | 57 | }
|
49 | 58 | sim = Simulation(start_date=start_date, seed=4, log_config=log_config)
|
50 | 59 |
|
51 | 60 | # Register the appropriate modules
|
52 | 61 | sim.register(demography.Demography(resourcefilepath=resourcefilepath),
|
| 62 | + care_of_women_during_pregnancy.CareOfWomenDuringPregnancy(resourcefilepath=resourcefilepath), |
| 63 | + contraception.Contraception(resourcefilepath=resourcefilepath), |
53 | 64 | enhanced_lifestyle.Lifestyle(resourcefilepath=resourcefilepath),
|
54 | 65 | healthsystem.HealthSystem(resourcefilepath=resourcefilepath),
|
55 | 66 | symptommanager.SymptomManager(resourcefilepath=resourcefilepath),
|
56 | 67 | healthseekingbehaviour.HealthSeekingBehaviour(resourcefilepath=resourcefilepath),
|
57 | 68 | healthburden.HealthBurden(resourcefilepath=resourcefilepath),
|
58 |
| - simplified_births.SimplifiedBirths(resourcefilepath=resourcefilepath), |
59 |
| - bladder_cancer.BladderCancer(resourcefilepath=resourcefilepath), |
60 |
| - schisto.Schisto(resourcefilepath=resourcefilepath) |
| 69 | + labour.Labour(resourcefilepath=resourcefilepath), |
| 70 | + newborn_outcomes.NewbornOutcomes(resourcefilepath=resourcefilepath), |
| 71 | + pregnancy_supervisor.PregnancySupervisor(resourcefilepath=resourcefilepath), |
| 72 | + postnatal_supervisor.PostnatalSupervisor(resourcefilepath=resourcefilepath), |
| 73 | + bladder_cancer.BladderCancer(resourcefilepath=resourcefilepath) |
61 | 74 | )
|
62 | 75 |
|
63 | 76 | # Run the simulation and flush the logger
|
|
0 commit comments