Draft: Resolve "Update wave stage functionality to be able to stage by group"

REMOVE IF ALREADY PART OF ISSUE DESCRIPTION

Overview

Tasks

• Update docs if needed (workflow, runbooks, README, migration metrics, etc.)

Acceptance Criteria

• Existing UTs pass locally

Closes #479 (closed)