From e72f1a8a71ba0d9dab44a09178e7e532690b56f1 Mon Sep 17 00:00:00 2001 From: Anton Lozhkov Date: Wed, 21 Sep 2022 13:54:32 +0200 Subject: [PATCH] Add torchvision to training deps (#607) --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index a9022ea7..71e7de77 100644 --- 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