Skip to content

Commit ca228e1

Browse files
anth-volkanth-volk
committed
Bump US dataset aliases
1 parent 7adaea0 commit ca228e1

3 files changed

Lines changed: 16 additions & 16 deletions

File tree

‎projects/policyengine-api-simulation/src/modal/gateway/endpoints.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -41,12 +41,12 @@
4141
JOB_METADATA_DICT_NAME = "simulation-api-job-metadata"
4242
DATASET_URIS = {
4343
"us": {
44-
"enhanced_cps": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
45-
"enhanced_cps_2024": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
46-
"cps": "hf://policyengine/policyengine-us-data/cps_2023.h5@1.77.0",
47-
"cps_2023": "hf://policyengine/policyengine-us-data/cps_2023.h5@1.77.0",
48-
"pooled_cps": "hf://policyengine/policyengine-us-data/pooled_3_year_cps_2023.h5@1.77.0",
49-
"pooled_3_year_cps_2023": "hf://policyengine/policyengine-us-data/pooled_3_year_cps_2023.h5@1.77.0",
44+
"enhanced_cps": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
45+
"enhanced_cps_2024": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
46+
"cps": "hf://policyengine/policyengine-us-data/cps_2023.h5@1.110.12",
47+
"cps_2023": "hf://policyengine/policyengine-us-data/cps_2023.h5@1.110.12",
48+
"pooled_cps": "hf://policyengine/policyengine-us-data/pooled_3_year_cps_2023.h5@1.110.12",
49+
"pooled_3_year_cps_2023": "hf://policyengine/policyengine-us-data/pooled_3_year_cps_2023.h5@1.110.12",
5050
},
5151
"uk": {
5252
"enhanced_frs": "hf://policyengine/policyengine-uk-data-private/enhanced_frs_2023_24.h5@1.40.3",

‎projects/policyengine-api-simulation/tests/gateway/test_endpoints.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ def test__given_submission_with_data__then_returns_resolved_bundle_metadata(
250250
"country": "us",
251251
"scope": "macro",
252252
"reform": {},
253-
"data": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
253+
"data": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
254254
}
255255

256256
# When
@@ -262,7 +262,7 @@ def test__given_submission_with_data__then_returns_resolved_bundle_metadata(
262262
assert data["resolved_app_name"] == "policyengine-simulation-us1-500-0-uk2-66-0"
263263
assert data["policyengine_bundle"] == {
264264
"model_version": "1.500.0",
265-
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
265+
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
266266
}
267267

268268
def test__given_submission_with_alias_data__then_bundle_dataset_stays_unresolved(
@@ -286,7 +286,7 @@ def test__given_submission_with_alias_data__then_bundle_dataset_stays_unresolved
286286
data = response.json()
287287
assert (
288288
data["policyengine_bundle"]["dataset"]
289-
== "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0"
289+
== "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12"
290290
)
291291

292292
def test__given_submission_with_uk_alias_data__then_bundle_dataset_is_versioned_uri(
@@ -341,7 +341,7 @@ def test__given_submission_with_runtime_bundle__then_accepts_internal_provenance
341341
assert data["policyengine_bundle"] == {
342342
"model_version": "1.500.0",
343343
"data_version": "1.78.2",
344-
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
344+
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
345345
}
346346
assert mock_modal["func"].last_payload["data_version"] == "1.78.2"
347347
assert "_runtime_bundle" not in mock_modal["func"].last_payload
@@ -388,7 +388,7 @@ def test__given_submitted_job__then_job_status_includes_bundle_metadata(
388388
"country": "us",
389389
"scope": "macro",
390390
"reform": {},
391-
"data": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
391+
"data": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
392392
},
393393
)
394394

@@ -403,7 +403,7 @@ def test__given_submitted_job__then_job_status_includes_bundle_metadata(
403403
assert data["resolved_app_name"] == "policyengine-simulation-us1-500-0-uk2-66-0"
404404
assert data["policyengine_bundle"] == {
405405
"model_version": "1.500.0",
406-
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
406+
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
407407
}
408408

409409
def test__given_submitted_job_with_telemetry__then_polling_echoes_run_id(

‎projects/policyengine-api-simulation/tests/gateway/test_models.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def test_job_submit_response_creates_with_all_fields(self):
313313
"model_version": "1.459.0",
314314
"policyengine_version": None,
315315
"data_version": None,
316-
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
316+
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
317317
},
318318
}
319319

@@ -332,7 +332,7 @@ def test_job_submit_response_creates_with_all_fields(self):
332332
assert response.policyengine_bundle.model_version == "1.459.0"
333333
assert response.policyengine_bundle.policyengine_version is None
334334
assert response.policyengine_bundle.dataset == (
335-
"hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0"
335+
"hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12"
336336
)
337337

338338

@@ -417,7 +417,7 @@ def test_job_status_response_accepts_bundle_metadata(self):
417417
"model_version": "1.459.0",
418418
"policyengine_version": None,
419419
"data_version": None,
420-
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0",
420+
"dataset": "hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12",
421421
},
422422
)
423423

@@ -426,7 +426,7 @@ def test_job_status_response_accepts_bundle_metadata(self):
426426
)
427427
assert response.policyengine_bundle is not None
428428
assert response.policyengine_bundle.dataset == (
429-
"hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.77.0"
429+
"hf://policyengine/policyengine-us-data/enhanced_cps_2024.h5@1.110.12"
430430
)
431431

432432

0 commit comments

Comments
 (0)