Update hubverse datatable creation to work with new state_pop_df

CDCgov · Feb 5, 2025 · daac312 · daac312
1 parent b2b636f
commit daac312
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pipelines/hubverse_create_observed_data_tables.py b/pipelines/hubverse_create_observed_data_tables.py
@@ -44,7 +44,7 @@ def save_observed_data_tables(
                     first_training_date=datetime.date(2023, 1, 1),
                     state_pop_df=state_pop,
                 ),
-                ["US"] + [x for x in state_pop["abb"]],
+                state_pop.get_column("abb").to_list(),
             )
         ).filter(pl.col("disease") == disease)
         for disease in ["COVID-19", "Influenza", "RSV", "Total"]