Add retry to failing test (#2430)

openreview · Nov 21, 2024 · 0ccf60f · 0ccf60f
1 parent fcfc1f6
commit 0ccf60f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_arr_venue_v2.py b/tests/test_arr_venue_v2.py
@@ -3528,6 +3528,16 @@ def test_sae_ae_assignments(self, client, openreview_client, helpers, test_clien
         helpers.await_queue_edit(openreview_client, edit_id=existing_edges[-1].id)
 
         ## Fetch corresponding June submissions with details replies using reviewer client, check replies for official reviews
+        retries, MAX_RETRIES = 0, 10
+        retry = True
+        while retries < MAX_RETRIES and retry:
+            try:
+                reviewer_six_client.get_note(june_submissions[2].id, details='replies')
+                retry = True
+                time.sleep(2)
+            except Exception as e:
+                retry = False
+                break
         same_note = reviewer_six_client.get_note(june_submissions[1].id, details='replies')
         with pytest.raises(openreview.OpenReviewException, match=r'User Reviewer ARRSix does not have permission to see'):
             reviewer_six_client.get_note(june_submissions[2].id, details='replies')