diff --git a/setup.py b/setup.py
index d9c8dffd7d..648459f7f2 100644
--- a/setup.py
+++ b/setup.py
@@ -102,7 +102,7 @@ def _parse_requirements_file(file_path):
     "flaky~=3.7.0",
     "ndjson>=0.3.1",
     "wheel>=0.36.2",
-    "pytest>=6.0.0",
+    "pytest>=6.0.0,<8.1.0",
     "onnxruntime>=1.7.0",
     "flask>=1.0.0",
     "flask-cors>=3.0.0",