Skip to content
This repository was archived by the owner on Jul 2, 2026. It is now read-only.

Commit 4983d65

Browse files
authored
Merge pull request #966 from PolicyEngine/codex/h5-migration-pr6-selection-reindexing
Extract local H5 selection and cloning seams
2 parents a337014 + 4639871 commit 4983d65

15 files changed

Lines changed: 1888 additions & 269 deletions

File tree

changelog.d/965.added

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Extract local H5 clone selection, entity reindexing, and variable cloning into dedicated build-output seams.

policyengine_us_data/build_outputs/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,5 +5,6 @@
55
H5 output request construction, exact calibration geography loading,
66
fingerprinting, clone-weight shape contracts, worker partitioning, source
77
dataset snapshot contracts, worker input normalization, worker-bootstrap
8-
artifacts, and worker-scoped session and validation context setup.
8+
artifacts, worker-scoped session and validation context setup, microsimulation
9+
access helpers, clone selection, entity reindexing, and source-variable cloning.
910
"""

policyengine_us_data/build_outputs/fingerprinting.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from policyengine_us_data.pipeline_metadata import pipeline_node
1212

1313
from .geography_loader import CalibrationGeographyLoader
14+
from .simulation_access import calculate_variable_values
1415

1516
FingerprintScope = Literal["regional", "national"]
1617
ScopeFingerprint = str
@@ -409,4 +410,9 @@ def _infer_n_records(
409410
from policyengine_us import Microsimulation
410411

411412
simulation = Microsimulation(dataset=str(source_dataset_path))
412-
return int(len(simulation.calculate("household_id", map_to="household").values))
413+
household_ids = calculate_variable_values(
414+
simulation,
415+
"household_id",
416+
map_to="household",
417+
)
418+
return int(len(household_ids))

0 commit comments

Comments
 (0)