Merge pull request #218 from sarahyurick/rerunfailures

Rerun flaky PyTest
NVIDIA · Aug 28, 2024 · cd4c490 · cd4c490
2 parents 2ee9d8c + 53766af
commit cd4c490
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -42,6 +42,7 @@ jobs:
           pip install wheel cython
           pip install --no-cache-dir .
           pip install pytest
+          pip install pytest-rerunfailures
       - name: Run tests
         run: |
           python -m pytest -v --cpu
diff --git a/tests/test_download.py b/tests/test_download.py
@@ -124,6 +124,7 @@ def test_incorrect_snapshot_order_news(self):
             start_snapshot = "2021-10"
             urls = get_common_crawl_urls(start_snapshot, end_snapshot, news=True)
 
+    @pytest.mark.flaky(reruns=3)
     def test_uneven_common_crawl_range(self):
         start_snapshot = "2021-03"
         end_snapshot = "2021-11"