Added retry action for CI (#1873)

* Added retry action * added retry on error * Update .github/workflows/ci_cpu.yml * Update ci_cpu.yml * Update ci_gpu.yml Co-authored-by: Aaqib <maaquib@gmail.com> Co-authored-by: Mark Saroufim <marksaroufim@fb.com>
pytorch · Jan 19, 2023 · 2d1bd26 · 2d1bd26
1 parent 221da4e
commit 2d1bd26
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/.github/workflows/ci_cpu.yml b/.github/workflows/ci_cpu.yml
@@ -36,8 +36,13 @@ jobs:
         run: |
           python ts_scripts/install_dependencies.py --environment=dev
       - name: Torchserve Sanity
-        run: |
-          python torchserve_sanity.py
+        uses: nick-fields/retry@v2
+        with:
+          timeout_minutes: 60
+          max_attempts: 3
+          retry_on: error
+          command: |
+            python torchserve_sanity.py
       - name: mvn install - unix
         if: matrix.os != 'windows-latest'
         run: |

diff --git a/.github/workflows/ci_gpu.yml b/.github/workflows/ci_gpu.yml
@@ -40,7 +40,13 @@ jobs:
         run: |
           python ts_scripts/install_dependencies.py --environment=dev --cuda=cu102
       - name: Torchserve Sanity
-        run: python torchserve_sanity.py
+        uses: nick-fields/retry@v2
+        with:
+          timeout_minutes: 60
+          retry_on: error
+          max_attempts: 3
+          command: |
+            python torchserve_sanity.py
       - name: mvn install
         run: |
             cd serving-sdk/ && ./mvnw clean install -q && cd ../