chore: test support retries

.github/workflows/run_all_tests.py

@@ -21,7 +21,7 @@ def run_all_tests():
     failed_tests = []
     for test_file in test_files:
         print(f"Running {test_file} ...")
-        result = subprocess.run(["pytest", test_file])
+        result = subprocess.run(["pytest", "--reruns", "2", "--reruns-delay", "0", test_file])
         if result.returncode != 0:
             print(f"Test failed: {test_file}")
             failed_tests.append(test_file)

docs/source/developer/contribution_guide.rst

@@ -59,9 +59,8 @@ Running the Tests
 
 .. code-block:: shell
 
-   HF_TOKEN=$YOUR_HF_TOKEN pytest -v tests/flux/test_flux_memory.py
-   HF_TOKEN=$YOUR_HF_TOKEN pytest -v tests/flux --ignore=tests/flux/test_flux_memory.py
-   HF_TOKEN=$YOUR_HF_TOKEN pytest -v tests/sana
+   HF_TOKEN=$YOUR_HF_TOKEN pytest -v tests/flux/test_flux_examples.py
+   HF_TOKEN=$YOUR_HF_TOKEN python .github/workflows/run_all_tests.py
 
 .. note::
 

tests/requirements.txt

@@ -10,3 +10,4 @@ insightface
 opencv-python
 facexlib
 onnxruntime
+pytest-rerunfailures