diff --git a/pyproject.toml b/pyproject.toml
index d7599fa6f..a336cc52c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -37,6 +37,7 @@ test = [
     "pytest",
     "pytest_check",
     "pytest-mock",
+    "pytest-timeout",
     "torch",
     "torchvision",
     "tensorflow==1.15",