[Nemo CICD] timeouts fix (NVIDIA#9407)

* timeouts fix * timeouts fix
rohitrango · Jun 6, 2024 · 3d95c23 · 3d95c23
1 parent 601a2ec
commit 3d95c23
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml
@@ -100,6 +100,7 @@ jobs:
     uses: ./.github/workflows/_test_template.yml
     with:
       RUNNER: self-hosted-azure
+      TIMEOUT: 30
       SCRIPT: |
         NEMO_NUMBA_MINVER=0.53 pytest -m "not pleasefixme" --with_downloads
       IS_OPTIONAL: true
@@ -109,7 +110,7 @@ jobs:
     uses: ./.github/workflows/_test_template.yml
     with:
       RUNNER: self-hosted-azure-cpu
-      TIMEOUT: 80
+      TIMEOUT: 60
       SCRIPT: |
         CUDA_VISIBLE_DEVICES="" NEMO_NUMBA_MINVER=0.53 pytest -m "not pleasefixme" --cpu --with_downloads --relax_numba_compat
 
@@ -4897,13 +4898,13 @@ jobs:
     uses: ./.github/workflows/_test_template.yml
     with:
       RUNNER: self-hosted-azure
+      TIMEOUT: 20
       SCRIPT: |
         CUDA_VISIBLE_DEVICES=0 python examples/asr/speech_to_text_eval.py \
             pretrained_name=QuartzNet15x5Base-En  \
             dataset_manifest=/home/TestData/librispeech/librivox-dev-other.json \
             batch_size=64 \
             tolerance=0.1012
-      TIMEOUT: 20
       AFTER_SCRIPT: |
         rm -f examples/asr/evaluation_transcripts.json
 
@@ -5057,4 +5058,4 @@ jobs:
 
       - if: ${{ always() && steps.pipeline-conclusion.outputs.SUCCESS == 'false' }}
         run: |
-          exit 1
+          exit 1