Add torchvision to training deps (#607)

e72f1a8a · Anton Lozhkov · GitHub · 4f1c989f · e72f1a8a
Unverified Commit e72f1a8a authored Sep 21, 2022 by Anton Lozhkov Committed by GitHub Sep 21, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

setup.py setup.py +1 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -174,7 +174,7 @@ extras = {}
 extras["quality"] = ["black==22.8", "isort>=5.5.4", "flake8>=3.8.3", "hf-doc-builder"]
 extras["docs"] = ["hf-doc-builder"]
 extras["training"] = ["accelerate", "datasets", "tensorboard", "modelcards"]
-extras["test"] = ["datasets", "onnxruntime", "pytest", "pytest-timeout", "pytest-xdist", "scipy", "transformers"]
+extras["test"] = ["datasets", "onnxruntime", "pytest", "pytest-timeout", "pytest-xdist", "scipy", "torchvision", "transformers"]
 extras["torch"] = deps_list("torch")

 if os.name == "nt":  # windows