retry on timeout

argoproj-labs · May 17, 2024 · 16dc07d · 16dc07d
1 parent eac1ea6
commit 16dc07d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gordian/repo.py b/gordian/repo.py
@@ -126,7 +126,7 @@ def _set_target_branch(self, target_branch, source_branch=None):
             self.branch_name = f"refs/heads/{datetime.datetime.now().strftime('%Y-%m-%d-%H%M%S.%f')}"
             self.source_branch = self.target_ref
 
-    @retry(GithubException, tries=3, delay=1, backoff=2)
+    @retry((GithubException, TimeoutError), tries=3, delay=1, backoff=2)
     def _get_repo_contents(self, path):
         try:
             logger.debug(f'Fetching repo contents {path}...')
@@ -135,6 +135,8 @@ def _get_repo_contents(self, path):
             if e.status == 404:
                 raise e
             logger.info(f'Error fetching repo contents: {e}')
+        except TimeoutError as e:
+            logger.info(f'Error fetching repo contents: {e}')
 
     @retry(GithubException, tries=3, delay=1, backoff=2)
     def _make_branch(self):