update helpers for torch test

openvinotoolkit · Oct 29, 2024 · b298c62 · b298c62
1 parent 07f0a6b
commit b298c62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/cross_fw/shared/helpers.py b/tests/cross_fw/shared/helpers.py
@@ -39,7 +39,7 @@ def get_cli_dict_args(args):
 
 
 MAP_BACKEND_PACKAGES = {
-    "torch": ["torch"],
+    "torch": ["torch", "torchvision"],
     "openvino": ["openvino"],
     "onnx": ["onnx", "onnxruntime"],
     "tf": ["tensorflow"],