Skip to content

Commit 3e01eae

Browse files
authored
Merge pull request #522 from PolicyEngine/maria/modal_improvement
Modal CI run efficiency improvement
2 parents 6a90ecf + b0e539b commit 3e01eae

5 files changed

Lines changed: 406 additions & 69 deletions

File tree

Makefile

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -95,11 +95,7 @@ data: download
9595
python policyengine_us_data/datasets/cps/extended_cps.py
9696
python policyengine_us_data/datasets/cps/enhanced_cps.py
9797
python policyengine_us_data/datasets/cps/small_enhanced_cps.py
98-
# 12000: number of households our GPUs can handle (found via trial and error).
99-
# --top=99.5: include only top 0.5% (vs default 1%) to preserve
100-
# representation of lower-income households.
101-
# --seed=3526: reproducible stratified sampling.
102-
python policyengine_us_data/datasets/cps/local_area_calibration/create_stratified_cps.py 12000 --top=99.5 --seed=3526
98+
python policyengine_us_data/datasets/cps/local_area_calibration/create_stratified_cps.py
10399

104100
publish-local-area:
105101
python policyengine_us_data/datasets/cps/local_area_calibration/publish_local_area.py

changelog_entry.yaml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
- bump: minor
2+
changes:
3+
added:
4+
- Parallelization of data building steps.
5+
- Checkpointing mechanism to resume data builds and testing modules from last successful step in Modal runs.
6+
changed:
7+
- Removed duplicate run of test_local_area_calibration tests.
8+
- Baked correct defaults into create_stratified_cps.py, removing hardcoded args from Makefile and Modal build.
9+
- DRYed up sequential build path to iterate SCRIPT_OUTPUTS instead of a redundant list.
10+
- Added thread-safe locking around volume.commit() for parallel checkpoint safety.

0 commit comments

Comments
 (0)