Add ACA marketplace plan selection proxies

.gitignore

@@ -25,6 +25,11 @@ node_modules
 !policyengine_us_data/storage/social_security_aux.csv
 !policyengine_us_data/storage/SSPopJul_TR2024.csv
 !policyengine_us_data/storage/national_and_district_rents_2023.csv
+!policyengine_us_data/storage/calibration_targets/aca_marketplace_county_plan_behavior_2024.csv
+!policyengine_us_data/storage/calibration_targets/aca_marketplace_rating_area_price_menu_2024.csv
+!policyengine_us_data/storage/calibration_targets/aca_marketplace_state_low_cost_plan_net_premium_proxy_2024.csv
+!policyengine_us_data/storage/calibration_targets/aca_marketplace_state_metal_selection_2024.csv
+!policyengine_us_data/storage/calibration_targets/aca_marketplace_state_price_ratio_2024.csv
 docs/.ipynb_checkpoints/
 
 ## ACA PTC state-level uprating factors

policyengine_us_data/calibration/publish_local_area.py

@@ -35,6 +35,9 @@
     SIMPLE_TAKEUP_VARS,
     apply_block_takeup_to_arrays,
 )
+from policyengine_us_data.utils.marketplace_plan_selection import (
+    assign_marketplace_plan_selection,
+)
 
 CHECKPOINT_FILE = Path("completed_states.txt")
 CHECKPOINT_FILE_DISTRICTS = Path("completed_districts.txt")
@@ -539,6 +542,25 @@ def build_h5(
         for var_name, bools in takeup_results.items():
             data[var_name] = {time_period: bools}
 
+    # === Apply marketplace plan selection proxies ===
+    print("Assigning marketplace bronze-vs-benchmark selections...")
+    tax_unit_clone_ids = np.repeat(
+        np.arange(n_clones, dtype=np.int64),
+        entities_per_clone["tax_unit"],
+    )
+    original_hh_ids = household_ids[active_hh].astype(np.int64)
+    tax_unit_state_fips = clone_geo["state_fips"][tax_unit_clone_ids]
+    tax_unit_blocks = active_blocks[tax_unit_clone_ids]
+    tax_unit_hh_ids = original_hh_ids[tax_unit_clone_ids]
+    marketplace_plan = assign_marketplace_plan_selection(
+        entity_states=tax_unit_state_fips,
+        entity_blocks=tax_unit_blocks,
+        entity_hh_ids=tax_unit_hh_ids,
+        entity_clone_ids=tax_unit_clone_ids,
+    )
+    for var_name, values in marketplace_plan.items():
+        data[var_name] = {time_period: values}
+
     # === Write H5 ===
     with h5py.File(str(output_path), "w") as f:
         for variable, periods in data.items():