diff --git a/setup.py b/setup.py
index 4dfe881..6056457 100644
--- a/setup.py
+++ b/setup.py
@@ -29,6 +29,7 @@
             "flake8",
             "isort",
             "pytest",
+            "pytest-rerunfailures",
             "pyyaml",
             "torchtestcase",
             "UMNN",