diff --git a/CHANGELOG.rst b/CHANGELOG.rst index ca882d26..e9fd0f24 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -1,3 +1,7 @@ +**5.0.3 - 07/12/24** + + - Fix bug in processing year ID of disability weight + **5.0.2 - 07/01/24** - Fix bug in validation of the year parameter diff --git a/src/vivarium_inputs/extract.py b/src/vivarium_inputs/extract.py index 33530bef..25fcb4fb 100644 --- a/src/vivarium_inputs/extract.py +++ b/src/vivarium_inputs/extract.py @@ -249,7 +249,9 @@ def extract_disability_weight( data["key"] = 1 year_df["key"] = 1 # Merge to get the Cartesian product, then drop the key column - data = pd.merge(data, year_df, on="key").drop("key", axis=1) + data = pd.merge(data.drop("year_id", axis=1), year_df, on="key").drop( + "key", axis=1 + ) else: data["year_id"] = year_id return data