vivarium-public-health 4.3.11__py3-none-any.whl → 4.3.12__py3-none-any.whl

vivarium_public_health/_version.py

@@ -1 +1 @@
-__version__ = "4.3.11"
+__version__ = "4.3.12"

vivarium_public_health/results/stratification.py

@@ -62,15 +62,14 @@ class ResultsStratifier(Component):
             is_vectorized=True,
             requires_columns=["current_time"],
         )
-        # TODO [MIC-4232]: simulants occasionally have event year of end_year_year+1 if the end time plus step size
-        #  lands in the next year. possible solution detailed in ticket
-        # builder.results.register_stratification(
-        #     "event_year",
-        #     [str(year) for year in range(self.start_year, self.end_year + 1)],
-        #     mapper=self.map_year,
-        #     is_vectorized=True,
-        #     requires_columns=["event_time"],
-        # )
+        builder.results.register_stratification(
+            "event_year",
+            [str(year) for year in range(self.start_year, self.end_year + 2)],
+            excluded_categories=[str(self.end_year + 1)],
+            mapper=self.map_year,
+            is_vectorized=True,
+            requires_columns=["event_time"],
+        )
         # TODO [MIC-3892]: simulants occasionally have entrance year of start_year-1 if the start time minus step size
         #  lands in the previous year. possible solution detailed in ticket
         # builder.results.register_stratification(

{vivarium_public_health-4.3.11.dist-info → vivarium_public_health-4.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vivarium_public_health
-Version: 4.3.11
+Version: 4.3.12
 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-4.3.11.dist-info → vivarium_public_health-4.3.12.dist-info}/RECORD

@@ -1,6 +1,6 @@
 vivarium_public_health/__about__.py,sha256=RgWycPypKZS80TpSX7o41cREnG8PfguNHDHLuLyl820,487
 vivarium_public_health/__init__.py,sha256=GDeeP-7OlCBwPuv_xQoB1wNmvCaFsqfTB7qnnYApm0w,1343
-vivarium_public_health/_version.py,sha256=FTMATci8OA-tNFnEujq_JNE-wHVYlo4B0glOFoqHJfs,23
+vivarium_public_health/_version.py,sha256=zEQ67gb4Xf_9S7gZ75IewB6b_Uhu_8Uw-gHHbGCjkZU,23
 vivarium_public_health/utilities.py,sha256=QNXQ6fhAr1HcV-GwKw7wQLz6QyuNxqNvMA-XujKjTgs,3035
 vivarium_public_health/disease/__init__.py,sha256=VUJHDLlE6ngo2qHNQUtZ8OWH5H_T7_ao-xsYKDkRmHw,443
 vivarium_public_health/disease/exceptions.py,sha256=vb30IIV82OiDf2cNZCs_E2rF6mdDDHbnZSND60no5CU,97
@@ -31,7 +31,7 @@ vivarium_public_health/results/mortality.py,sha256=9L8tphHn2BEt5sSDkvSfcAyyhsGA_
 vivarium_public_health/results/observer.py,sha256=cyltdWY6mowUUSc8fHDWFkqJJGEFnwhcI4XB98RgJfo,7603
 vivarium_public_health/results/risk.py,sha256=bPZq2El00MZ5A6ttqA-wvbHLm4E3OZcEZOkqV_GENk4,5838
 vivarium_public_health/results/simple_cause.py,sha256=ibdE6KwhDfQWntCVkOEooBcmUydEoupmd3_poHSHyu8,1007
-vivarium_public_health/results/stratification.py,sha256=4I3YGHVabNAZENE7YboOtWsWU4X-8LUBJ9iwYMbpl6E,5387
+vivarium_public_health/results/stratification.py,sha256=Ok0I_Vcbq6c56EXeCKjIyprwTZxAwmddx5zlTf-KOg8,5243
 vivarium_public_health/risks/__init__.py,sha256=z8DcnZGxqNVAyFZm2WAV-IVNGvrSS4izju_0DNe2Ghw,212
 vivarium_public_health/risks/base_risk.py,sha256=XrZ5iytFabPS2_9-jI0dZQKK206yy_ZvCbReUoAVrAQ,10851
 vivarium_public_health/risks/data_transformations.py,sha256=SgdPKc95BBqgMNUdlAQM8k6iaXcpxnjk5B2ySTES1Yg,9269
@@ -43,8 +43,8 @@ vivarium_public_health/treatment/__init__.py,sha256=wONElu9aJbBYwpYIovYPYaN_GYfV
 vivarium_public_health/treatment/magic_wand.py,sha256=zg4I48G-l9fC6-qjvApbM1zNACJimZlX9ZZ9fdHKwvc,1985
 vivarium_public_health/treatment/scale_up.py,sha256=_bsf9c_yVc7-q_-yBcXimISTUfYzPps1auH0uEf7sfQ,7048
 vivarium_public_health/treatment/therapeutic_inertia.py,sha256=ZIHnpuszZwA_BkS53JbSLvpcnX_bqG2knwCUyUgkA9M,2362
-vivarium_public_health-4.3.11.dist-info/licenses/LICENSE.txt,sha256=mN4bNLUQNcN9njYRc_3jCZkfPySVpmM6MRps104FxA4,1548
-vivarium_public_health-4.3.11.dist-info/METADATA,sha256=-QvvuE7m66YSkjnBa1GdUrK3bq9U3KlDlGvHxYQ0AvU,3885
-vivarium_public_health-4.3.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vivarium_public_health-4.3.11.dist-info/top_level.txt,sha256=VVInlpzCFD0UNNhjOq_j-a29odzjwUwYFTGfvqbi4dY,23
-vivarium_public_health-4.3.11.dist-info/RECORD,,
+vivarium_public_health-4.3.12.dist-info/licenses/LICENSE.txt,sha256=mN4bNLUQNcN9njYRc_3jCZkfPySVpmM6MRps104FxA4,1548
+vivarium_public_health-4.3.12.dist-info/METADATA,sha256=pJRU6DuPtz1vqX7ORdKb8wy-HpbqnWsRmx7SRYwWyK0,3885
+vivarium_public_health-4.3.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vivarium_public_health-4.3.12.dist-info/top_level.txt,sha256=VVInlpzCFD0UNNhjOq_j-a29odzjwUwYFTGfvqbi4dY,23
+vivarium_public_health-4.3.12.dist-info/RECORD,,