From d1490602e404bf88eab769068a5fb4f45604c893 Mon Sep 17 00:00:00 2001 From: Jonas Heinicke <152563145+Jonasjjj96@users.noreply.github.com> Date: Tue, 17 Sep 2024 10:56:19 +0200 Subject: [PATCH] Fix: Update getPopulationData.py to account for 2023 population data Fix: Mismatch in total population due to 2023 data update The newly released population data for 2023 contains only data from 2023, causing a mismatch in the total population when compared with previous years. This results in an error being thrown during the population data validation process. --- pycode/memilio-epidata/memilio/epidata/getPopulationData.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pycode/memilio-epidata/memilio/epidata/getPopulationData.py b/pycode/memilio-epidata/memilio/epidata/getPopulationData.py index 844ba54475..ed8baa447c 100644 --- a/pycode/memilio-epidata/memilio/epidata/getPopulationData.py +++ b/pycode/memilio-epidata/memilio/epidata/getPopulationData.py @@ -206,10 +206,13 @@ def test_total_population(df_pop, age_cols): total_sum_2020 = 83155031 total_sum_2021 = 83237124 total_sum_2022 = 84358845 + total_sum_2023 = 84669326 total_sum = df_pop[age_cols].sum().sum() - if total_sum == total_sum_2022: + if total_sum == total_sum_2023: pass + elif total_sum == total_sum_2022: + warnings.warn('Using data of 2022. Newer data is available.') elif total_sum == total_sum_2021: warnings.warn('Using data of 2021. Newer data is available.') elif total_sum == total_sum_2020: