From 7f0340f7b8f3930d48eb976e1ea610a482a378f8 Mon Sep 17 00:00:00 2001
From: = <stolld@cs.uni-freiburg.de>
Date: Mon, 1 Jul 2024 03:26:36 +0200
Subject: [PATCH] add torchvision to non-dev dependencies

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index af33458a..8ac2e386 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -63,6 +63,7 @@ seaborn = "^0.13"
 pyyaml = "^6"
 tensorboard = "^2"
 typing-extensions = "*"
+torchvision = ">=0.8.0"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.4"