Retry download of openfl-nightly package after its publish (#1586)

* Added retry for download package step

Signed-off-by: noopur <[email protected]>

* Added retry for download package step

Signed-off-by: noopur <[email protected]>

* Use variables for max attempt and wait time

Signed-off-by: noopur <[email protected]>

---------

Signed-off-by: noopur <[email protected]>

Author: noopurintel

.github/workflows/publish_nightly_package.yml

@@ -89,10 +89,27 @@ jobs:
       id: install_package_using_wheel
       run: |
         python3 -m pip uninstall -y openfl openfl-nightly
+        # Install the package with retry logic
+        MAX_ATTEMPTS=5
+        WAIT_TIME_IN_SEC=20
         if [ "${{ env.TEST_PYPI }}" == "true" ]; then
-          python3 -m pip install -i https://test.pypi.org/simple/ openfl-nightly==${{ env.NEW_VERSION }}
+          PIP_CMD="python3 -m pip install -i https://test.pypi.org/simple/ openfl-nightly==${{ env.NEW_VERSION }}"
         else
-          python3 -m pip install openfl-nightly==${{ env.NEW_VERSION }}
+          PIP_CMD="python3 -m pip install openfl-nightly==${{ env.NEW_VERSION }}"
+        fi
+        for i in $(seq 1 $MAX_ATTEMPTS); do
+          if $PIP_CMD; then
+            echo "Successfully installed openfl-nightly using command '$PIP_CMD'"
+            break
+          else
+            echo "Attempt $i/$MAX_ATTEMPTS failed. Retrying in $WAIT_TIME_IN_SEC seconds..."
+            sleep $WAIT_TIME_IN_SEC
+          fi
+        done
+        # Check if the package was installed successfully
+        if ! python3 -m pip show openfl-nightly > /dev/null; then
+          echo "Failed to install openfl-nightly after $MAX_ATTEMPTS attempts."
+          exit 1
         fi
         python3 -m pip install -r test-requirements.txt
 
@@ -108,4 +125,4 @@ jobs:
       run: |
         echo "**Version:** $(pip list | grep openfl)" >> $GITHUB_STEP_SUMMARY
         echo "**Commit ID:** ${{ env.COMMIT_ID }}" >> $GITHUB_STEP_SUMMARY
-        echo "Package information printed successfully!"
+        echo "Package information printed successfully!"