From ec65bfa1d2d736c536976fc0dcbe11e7b83b3694 Mon Sep 17 00:00:00 2001
From: Fam Zheng <fam@euphon.net>
Date: Sun, 23 Nov 2025 18:24:36 +0000
Subject: [PATCH] Add torch and torchvision dependencies, ensure venv is
 activated in script section

---
 .gitlab-ci.yml   | 1 +
 requirements.txt | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 7047383..8bbc5b4 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -91,6 +91,7 @@ zy:
     - source venv/bin/activate
     - uv pip install -r requirements.txt --default-index https://pypi.tuna.tsinghua.edu.cn/simple
   script:
+    - source venv/bin/activate
     - make fetch-quick
     - make sbs-quick
     - make train-quick
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 37234a1..d5acc0a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -19,6 +19,8 @@ matplotlib
 scikit-learn
 scipy
 seaborn
+torch
+torchvision
 kornia
 opencv-python
 opencv-contrib-python