CI RF-tests, try TF install several times

pypa/pip#4796
rwth-i6 · Jul 12, 2024 · d5b954b · d5b954b
1 parent 65e2623
commit d5b954b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -410,7 +410,10 @@ jobs:
           pip install --user --upgrade "gast<=0.4.0"
         fi
 
-        pip install --user --upgrade --progress-bar=off tensorflow==${{ matrix.tf-version }}
+        # Retry several times in case download breaks. https://github.com/pypa/pip/issues/4796
+        for ((i=0; i<3; i++)); do
+          if pip install --user --upgrade --progress-bar=off tensorflow==${{ matrix.tf-version }}; then break; fi
+        done
 
     - name: Test Python/Numpy/PyTorch versions.
       run: |