From b298c622ab9b213e611a8d46ea0a3536a4b0b001 Mon Sep 17 00:00:00 2001
From: anzr299 <aamir.nazir@intel.com>
Date: Tue, 29 Oct 2024 14:11:08 +0400
Subject: [PATCH] update helpers for torch test

---
 tests/cross_fw/shared/helpers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/cross_fw/shared/helpers.py b/tests/cross_fw/shared/helpers.py
index 1b9b9ef2d66..8c9306139b2 100644
--- 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"],