Merge pull request #2216 from PolicyEngine/fix/public-uk-data

Use new public Hugging Face model
PolicyEngine · Feb 28, 2025 · 04dbdfc · 04dbdfc
2 parents 2edf784 + ba3a5e6
commit 04dbdfc
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 5 deletions.
diff --git a/changelog_entry.yaml b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: patch
+  changes:
+    changed:
+    - Altered UK constituency data addresses to use new public Hugging Face model
diff --git a/policyengine_api/country.py b/policyengine_api/country.py
@@ -67,7 +67,7 @@ def build_microsimulation_options(self) -> dict:
         options = dict()
         if self.country_id == "uk":
             constituency_names_path = download_huggingface_dataset(
-                repo="policyengine/policyengine-uk-data",
+                repo="policyengine/policyengine-uk-data-public",
                 repo_filename="constituencies_2024.csv",
             )
             constituency_names = pd.read_csv(constituency_names_path)

diff --git a/policyengine_api/endpoints/economy/compare.py b/policyengine_api/endpoints/economy/compare.py
@@ -570,7 +570,7 @@ def uk_constituency_breakdown(
     reform_hnet = reform["household_net_income"]
 
     constituency_weights_path = download_huggingface_dataset(
-        repo="policyengine/policyengine-uk-data",
+        repo="policyengine/policyengine-uk-data-public",
         repo_filename="parliamentary_constituency_weights.h5",
     )
     with h5py.File(constituency_weights_path, "r") as f:
@@ -579,7 +579,7 @@ def uk_constituency_breakdown(
         ]  # {2025: array(650, 100180) where cell i, j is the weight of household record i in constituency j}
 
     constituency_names_path = download_huggingface_dataset(
-        repo="policyengine/policyengine-uk-data",
+        repo="policyengine/policyengine-uk-data-public",
         repo_filename="constituencies_2024.csv",
     )
     constituency_names = pd.read_csv(

diff --git a/policyengine_api/jobs/calculate_economy_simulation_job.py b/policyengine_api/jobs/calculate_economy_simulation_job.py
@@ -270,11 +270,11 @@ def _create_simulation_uk(
         simulation.default_calculation_period = time_period
         if region != "uk":
             constituency_weights_path = download_huggingface_dataset(
-                repo="policyengine/policyengine-uk-data",
+                repo="policyengine/policyengine-uk-data-public",
                 repo_filename="parliamentary_constituency_weights.h5",
             )
             constituency_names_path = download_huggingface_dataset(
-                repo="policyengine/policyengine-uk-data",
+                repo="policyengine/policyengine-uk-data-public",
                 repo_filename="constituencies_2024.csv",
             )
             constituency_names = pd.read_csv(constituency_names_path)