From e9d3684614a58a64b59523832cf07ad4d300b266 Mon Sep 17 00:00:00 2001 From: Anthony Volk Date: Thu, 11 Dec 2025 14:22:20 +0400 Subject: [PATCH 1/2] fix: Don't set dataset version for US simulations; allow .py to handle this --- policyengine_api/data/model_setup.py | 7 +------ tests/fixtures/services/economy_service.py | 2 +- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/policyengine_api/data/model_setup.py b/policyengine_api/data/model_setup.py index 960e7de76..b734008e6 100644 --- a/policyengine_api/data/model_setup.py +++ b/policyengine_api/data/model_setup.py @@ -1,5 +1,3 @@ -from policyengine_api.utils.hugging_face import get_latest_commit_tag - ENHANCED_FRS = "hf://policyengine/policyengine-uk-data/enhanced_frs_2023_24.h5" FRS = "hf://policyengine/policyengine-uk-data/frs_2023_24.h5" @@ -30,10 +28,7 @@ def get_dataset_version(country_id: str) -> str | None: case "uk": return None case "us": - return get_latest_commit_tag( - repo_id="policyengine/policyengine-us-data", - repo_type="model", - ) + return None case _: raise ValueError(f"Unknown country ID: {country_id}") diff --git a/tests/fixtures/services/economy_service.py b/tests/fixtures/services/economy_service.py index a02e5dc85..de043853e 100644 --- a/tests/fixtures/services/economy_service.py +++ b/tests/fixtures/services/economy_service.py @@ -17,7 +17,7 @@ MOCK_EXECUTION_ID = "mock_execution_id_12345" MOCK_PROCESS_ID = "job_20250626120000_1234" MOCK_MODEL_VERSION = "1.2.3" -MOCK_DATA_VERSION = "2024.1.0" +MOCK_DATA_VERSION = None MOCK_REFORM_POLICY_JSON = json.dumps( {"sample_param": {"2024-01-01.2100-12-31": 15}} From 2d17e71f9f3130d492c356383091a56cdc2d9ef9 Mon Sep 17 00:00:00 2001 From: Anthony Volk Date: Thu, 11 Dec 2025 14:25:59 +0400 Subject: [PATCH 2/2] chore: Changelog --- changelog_entry.yaml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/changelog_entry.yaml b/changelog_entry.yaml index e69de29bb..26fadaa48 100644 --- a/changelog_entry.yaml +++ b/changelog_entry.yaml @@ -0,0 +1,4 @@ +- bump: patch + changes: + removed: + - Explicit setting of US dataset version in economy service \ No newline at end of file