fix: wait for medium e2e to finish and pass

Signed-off-by: Charlie Doern <[email protected]>

Author: cdoern

.github/mergify.yml

@@ -23,7 +23,25 @@ pull_request_rules:
       - and:
         - -files~=^\.github/(actions|workflows)/.*\.ya?ml$
         - -files~=^\.github/workflows/actionlint\.
-
+    # medium e2e workflow
+    - or:
+      - and:
+        # note this should match the triggering criteria in 'e2e-nvidia-l4-x1.yml'
+        - check-success~=e2e-medium-workflow-complete
+        - or:
+          - files~=\.py$
+          - files=pyproject.toml
+          - files~=^requirements.*\.txt$
+          - files=tox.ini
+          - files~=^scripts/[^/]+\.sh$
+          - files=.github/workflows/lint.yml
+      - and:
+        - -files~=\.py$
+        - -files=pyproject.toml
+        - -files~=^requirements.*\.txt$
+        - -files=tox.ini
+        - -files~=^scripts/[^/]+\.sh$
+        - -files=.github/workflows/lint.yml
     # code lint workflow
     - or:
       - and:

.github/workflows/e2e-nvidia-l40s-x4-sdk.yml

@@ -324,4 +324,15 @@ jobs:
         if: steps.phase-2-upload-s3.outcome == 'failure'
         run: |
           echo "::warning::Failed to upload Phase 2 loss graph to S3. This won't block the workflow, but you may want to investigate."
-          echo "Loss graph upload failed" >> "${GITHUB_STEP_SUMMARY}"
+          echo "Loss graph upload failed" >> "${GITHUB_STEP_SUMMARY}"
+
+  e2e-medium-workflow-complete:
+    # we don't want to block PRs on failed EC2 cleanup
+    # so not requiring "stop-medium-ec2-runner" as well
+    permissions:
+      checks: read
+    uses: ./.github/workflows/status-checks.yml
+    with:
+      job_ids: >- # Space-separated job ids to wait on for status checks
+        start-large-ec2-runner
+        e2e-medium-test