From f765f94230b3bf1ff2d005ef504bad07926a17b7 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Thu, 13 Apr 2023 10:02:13 +0000
Subject: [PATCH] Reintroduce torchvision requirements as it's needed for
 training

---
 training-requirements.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/training-requirements.txt b/training-requirements.txt
index dea26c5..08ca60d 100644
--- a/training-requirements.txt
+++ b/training-requirements.txt
@@ -4,4 +4,5 @@ mlflow==2.2.1
 Shapely==2.0.1
 teklia-toolbox==0.1.3
 tensorboard==2.12.1
+torchvision==0.15.1
 tqdm==4.65.0
-- 
GitLab