Skip to content

Commit e72f1a8

Browse files
authored
Add torchvision to training deps (#607)
1 parent 4f1c989 commit e72f1a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def run(self):
174174
extras["quality"] = ["black==22.8", "isort>=5.5.4", "flake8>=3.8.3", "hf-doc-builder"]
175175
extras["docs"] = ["hf-doc-builder"]
176176
extras["training"] = ["accelerate", "datasets", "tensorboard", "modelcards"]
177-
extras["test"] = ["datasets", "onnxruntime", "pytest", "pytest-timeout", "pytest-xdist", "scipy", "transformers"]
177+
extras["test"] = ["datasets", "onnxruntime", "pytest", "pytest-timeout", "pytest-xdist", "scipy", "torchvision", "transformers"]
178178
extras["torch"] = deps_list("torch")
179179

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

0 commit comments

Comments
 (0)