Fix e2e tests on AWS

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Author: chtruong814

.github/workflows/cicd-main.yml

@@ -170,7 +170,7 @@ jobs:
             runner: ${{ needs.pre-flight.outputs.runner_prefix }}-gpu-x2
           - script: L2_TRTLLM_API_Deploy_Query
             runner: ${{ needs.pre-flight.outputs.runner_prefix }}-gpu-x2
-    needs: [cicd-unit-tests-trtllm]
+    needs: [cicd-unit-tests-trtllm, pre-flight]
     runs-on: ${{ matrix.runner }}
     name: ${{ matrix.is_optional && 'PLEASEFIXME_' || '' }}${{ matrix.script }}
     if: |
@@ -201,7 +201,7 @@ jobs:
           runner: ${{ matrix.runner }}
 
   cicd-e2e-tests-trt-onnx:
-    needs: [cicd-unit-tests-trtllm]
+    needs: [cicd-unit-tests-trtllm, pre-flight]
     runs-on: ${{ needs.pre-flight.outputs.runner_prefix }}-gpu-x2
     name: ${{ matrix.is_optional && 'PLEASEFIXME_' || '' }}${{ matrix.script }}
     environment: nemo-ci
@@ -234,7 +234,7 @@ jobs:
           runner: ${{ matrix.runner }}
 
   cicd-e2e-tests-vllm:
-    needs: [cicd-unit-tests-vllm]
+    needs: [cicd-unit-tests-vllm, pre-flight]
     runs-on: ${{ needs.pre-flight.outputs.runner_prefix }}-gpu-x2
     name: ${{ matrix.is_optional && 'PLEASEFIXME_' || '' }}${{ matrix.script }}
     environment: nemo-ci
@@ -265,7 +265,7 @@ jobs:
           runner: ${{ matrix.runner }}
 
   cicd-e2e-tests-inframework:
-    needs: [cicd-unit-tests-trtllm, cicd-unit-tests-vllm]
+    needs: [cicd-unit-tests-trtllm, cicd-unit-tests-vllm, pre-flight]
     runs-on: ${{ needs.pre-flight.outputs.runner_prefix }}-gpu-x2
     name: ${{ matrix.is_optional && 'PLEASEFIXME_' || '' }}${{ matrix.script }}
     environment: nemo-ci