[testing] add timeout_decorator (#3543)

2020-05-01 09:05:47 -04:00
parent b8686174be
commit 18db92dd9a
3 changed files with 5 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -72,7 +72,7 @@ extras["torch"] = ["torch==1.4.0"]
 extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"]
 extras["all"] = extras["serving"] + ["tensorflow", "torch"]

-extras["testing"] = ["pytest", "pytest-xdist"]
+extras["testing"] = ["pytest", "pytest-xdist", "timeout-decorator"]
 extras["docs"] = ["recommonmark", "sphinx", "sphinx-markdown-tables", "sphinx-rtd-theme"]
 extras["quality"] = [
    "black",