From d5b954b8f6e4c84ec2c289733590e1bf4154ba8b Mon Sep 17 00:00:00 2001
From: Albert Zeyer <zeyer@i6.informatik.rwth-aachen.de>
Date: Fri, 12 Jul 2024 16:40:51 +0200
Subject: [PATCH] CI RF-tests, try TF install several times

https://github.com/pypa/pip/issues/4796
---
 .github/workflows/main.yml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 146a4cf4f..95210f49d 100644
--- 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: |