Add pytest-subtests to the dependencies

As discussed in [1], pytest-subtests changes behavior of .skipTest() having effect to really skip individual subtests or skip the entire test if module is not installed. Huggingface Accelerate has module in its dependencies. It makes sense to add it for Transformers as well to avoid divergent environment between users and ci. See[1]: huggingface#34723 (comment) Signed-off-by: Dmitry Rogozhkin <[email protected]>
dvrogozh · Nov 15, 2024 · f68c9d1 · f68c9d1
1 parent a3d69a8
commit f68c9d1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -148,6 +148,7 @@
     "pyyaml>=5.1",
     "pydantic",
     "pytest>=7.2.0,<8.0.0",
+    "pytest-subtests",
     "pytest-timeout",
     "pytest-xdist",
     "python>=3.9.0",
@@ -320,6 +321,7 @@ def run(self):
     deps_list(
         "pytest",
         "pytest-rich",
+        "pytest-subtests",
         "pytest-xdist",
         "timeout-decorator",
         "parameterized",