vivarium-public-health 2.3.1__tar.gz → 2.3.2__tar.gz

{vivarium_public_health-2.3.1 → vivarium_public_health-2.3.2}/CHANGELOG.rst

@@ -1,3 +1,7 @@
+**2.3.2 - 3/13/24**
+
+ - Update to Mortality Observer to make subclassing easier.
+
 **2.3.1 - 3/11/24**
 
  - Update Mortality Observer to include tracked in population filter

{vivarium_public_health-2.3.1 → vivarium_public_health-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vivarium_public_health
-Version: 2.3.1
+Version: 2.3.2
 Summary: Components for modelling diseases, risks, and interventions with ``vivarium``
 Home-page: https://github.com/ihmeuw/vivarium_public_health
 Author: The vivarium developers

vivarium_public_health-2.3.2/src/vivarium_public_health/_version.py

@@ -0,0 +1 @@
+__version__ = "2.3.2"

{vivarium_public_health-2.3.1 → vivarium_public_health-2.3.2}/src/vivarium_public_health/metrics/mortality.py

@@ -54,6 +54,17 @@ class MortalityObserver(Component):
         }
     }
 
+    def __init__(self):
+        super().__init__()
+        self.causes_of_death = ["other_causes"]
+        self.required_death_columns = ["alive", "exit_time"]
+        self.required_yll_columns = [
+            "alive",
+            "cause_of_death",
+            "exit_time",
+            "years_of_life_lost",
+        ]
+
     ##############
     # Properties #
     ##############
@@ -75,18 +86,11 @@ class MortalityObserver(Component):
     def setup(self, builder: Builder) -> None:
         self.clock = builder.time.clock()
         self.config = builder.configuration.stratification.mortality
-        self._cause_components = builder.components.get_components_by_type(
+        cause_components = builder.components.get_components_by_type(
             (DiseaseState, RiskAttributableDisease)
         )
-        self.causes_of_death = ["other_causes"] + [
-            cause.state_id for cause in self._cause_components if cause.has_excess_mortality
-        ]
-        self.required_death_columns = ["alive", "exit_time"]
-        self.required_yll_columns = [
-            "alive",
-            "cause_of_death",
-            "exit_time",
-            "years_of_life_lost",
+        self.causes_of_death += [
+            cause.state_id for cause in cause_components if cause.has_excess_mortality
         ]
         if not self.config.aggregate:
             for cause_of_death in self.causes_of_death:

{vivarium_public_health-2.3.1 → vivarium_public_health-2.3.2}/src/vivarium_public_health.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vivarium_public_health
-Version: 2.3.1
+Version: 2.3.2
 Summary: Components for modelling diseases, risks, and interventions with ``vivarium``
 Home-page: https://github.com/ihmeuw/vivarium_public_health
 Author: The vivarium developers

vivarium_public_health-2.3.1/src/vivarium_public_health/_version.py

@@ -1 +0,0 @@
-__version__ = "2.3.1"