diff --git a/etl/steps/data/garden/who/2024-07-30/ghe.py b/etl/steps/data/garden/who/2024-07-30/ghe.py index 2934a61b406..43bb9e89d53 100644 --- 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 index 08812693d0d..944630885df 100644 --- a/lib/datautils/uv.lock +++ b/lib/datautils/uv.lock @@ -666,7 +666,7 @@ name = "importlib-metadata" version = "8.5.0" source = { registry = "https://pypi.org/simple" } dependencies = [ - { name = "zipp", marker = "python_full_version < '3.11'" }, + { name = "zipp", marker = "python_full_version < '3.10'" }, ] sdist = { url = "https://files.pythonhosted.org/packages/cd/12/33e59336dca5be0c398a7482335911a33aa0e20776128f038019f1a95f1b/importlib_metadata-8.5.0.tar.gz", hash = "sha256:71522656f0abace1d072b9e5481a48f07c138e00f079c38c8f883823f9c26bd7", size = 55304 } wheels = [ @@ -1113,7 +1113,7 @@ requires-dist = [ { name = "colorama", specifier = ">=0.4.4" }, { name = "gdown", specifier = ">=4.5.2" }, { name = "gsheets", specifier = ">=0.6.1" }, - { name = "pandas", specifier = "==2.2.1" }, + { name = "pandas", specifier = ">=2.2.1" }, { name = "pyarrow", specifier = ">=10.0.1" }, { name = "pydrive2", specifier = ">=1.15.0" }, { name = "structlog", specifier = ">=21.5.0" },