Skip to content

Expose Stage 1 substep status in pipeline status payloads#1115

Merged
anth-volk merged 2 commits into
mainfrom
fix/stage1-substep-status-ingestion
May 22, 2026
Merged

Expose Stage 1 substep status in pipeline status payloads#1115
anth-volk merged 2 commits into
mainfrom
fix/stage1-substep-status-ingestion

Conversation

@anth-volk
Copy link
Copy Markdown
Collaborator

Fixes #1114

Summary

  • Persist Stage 1 substep status events, finalized substep results, and the current Stage 1 substep under each run directory.
  • Wire build_datasets to attach the recorder when a run ID is present, committing status updates to the pipeline volume as they are written.
  • Expose stage_1_status in detailed pipeline status payloads and stage_1_current in pipeline run indexes.

Testing

  • make lint
  • uv run --no-sync ruff check modal_app/data_build.py modal_app/step_manifests/status.py policyengine_us_data/build_datasets/__init__.py policyengine_us_data/build_datasets/coordinator.py policyengine_us_data/build_datasets/status_store.py tests/unit/test_build_dataset_status_store.py tests/unit/test_pipeline_status.py tests/unit/test_modal_data_build.py
  • uv run --no-sync pytest tests/unit/test_build_dataset_status_store.py tests/unit/test_pipeline_status.py tests/unit/test_modal_data_build.py -q
  • uv run --no-sync --with pyyaml python scripts/run_quality_guards.py

Note: full uv run --no-sync ruff check . is currently blocked by pre-existing unrelated lint failures outside this change set.

@anth-volk anth-volk force-pushed the fix/stage1-substep-status-ingestion branch from 5bbe34a to 04bc94b Compare May 22, 2026 19:27
@anth-volk anth-volk force-pushed the fix/stage1-substep-status-ingestion branch from f9d8931 to b92054f Compare May 22, 2026 20:08
@anth-volk anth-volk marked this pull request as ready for review May 22, 2026 20:41
@anth-volk anth-volk merged commit a34eb3a into main May 22, 2026
13 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Expose Stage 1 substep status in pipeline status payloads

1 participant