From b8d08f82b736cffff169b6054c3846e4f3b7fd99 Mon Sep 17 00:00:00 2001
From: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
Date: Tue, 16 Sep 2025 17:17:52 +0530
Subject: [PATCH] Move torchvision dependency to optional dev-test group

Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 46abccac9..67bf114ae 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,6 @@
     "safetensors",
     "torch>=2.6",
     "torchprofile>=0.0.4",
-    "torchvision",
 ]
 
 optional_deps = {
@@ -79,6 +78,7 @@
         "pytest-cov",
         "pytest-timeout",
         "timm",
+        "torchvision",
         "tox>4.18",
         "tox-current-env>=0.0.12",
     ],