🔨 use categoricals instead of string for GHE (#3604)

* 🔨 use categoricals instead of string for GHE
owid · Nov 25, 2024 · ee7acda · ee7acda
1 parent 4d4fb05
commit ee7acda
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/etl/steps/data/garden/who/2024-07-30/ghe.py b/etl/steps/data/garden/who/2024-07-30/ghe.py
@@ -71,7 +71,7 @@
 def run(dest_dir: str) -> None:
     # read dataset from meadow
     ds_meadow = paths.load_dataset()
-    tb = ds_meadow.read("ghe")
+    tb = ds_meadow.read("ghe", safe_types=False)
     tb = tb.drop(columns="flag_level")
 
     tb = rename_table_for_compatibility(tb)
@@ -114,7 +114,7 @@ def run(dest_dir: str) -> None:
 
 def rename_table_for_compatibility(tb: Table) -> Table:
     """Rename columns and labels to be compatible with the previous version of the dataset."""
-    tb.age_group = tb.age_group.map(AGE_GROUPS_MAP)
+    tb.age_group = dataframes.map_series(tb.age_group, AGE_GROUPS_MAP)
     tb = tb.rename(
         columns={
             "val_dths_count_numeric": "death_count",

diff --git a/lib/datautils/uv.lock b/lib/datautils/uv.lock