diff --git a/pyproject.toml b/pyproject.toml
index 77e88f320..85c65d637 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -289,7 +289,6 @@ dependencies = [
     "pydantic-settings",
     "requests",
     "requests-mock",
-    "torch",
     "torchvision",
     'tzdata; sys_platform == "win32"',
     "vws-auth-tools",
@@ -300,7 +299,7 @@ dev = [
     "actionlint-py==1.6.27.13",
     "check-manifest==0.49",
     "check-wheel-contents==0.6.0",
-    "deptry==0.16.0",
+    "deptry==0.16.1",
     "dirty-equals==0.7.1.post0",
     "doc8==1.1.1",
     "docker==7.0.0",
@@ -331,6 +330,7 @@ dev = [
     "sphinxcontrib-spelling==8",
     "sybil==6.0.3",
     "tenacity==8.2.3",
+    "torch==2.2.2",
     "types-Pillow==10.2.0.20240406",
     "types-PyYAML==6.0.12.20240311",
     "types-requests==2.31.0.20240406",