diff --git a/setup.py b/setup.py
index 997cf369cb4a78..f1c53db352ef4d 100644
--- a/setup.py
+++ b/setup.py
@@ -310,6 +310,7 @@ def run(self):
     deps_list(
         "pytest",
         "pytest-xdist",
+        "timeout-decorator",
         "parameterized",
         "psutil",
         "datasets",