From de1341c5753ae509db1e9a51a856787b869fcd96 Mon Sep 17 00:00:00 2001
From: "Hafiz, Aqib (PG/T - Comp Sci & Elec Eng)" <ah02821@surrey.ac.uk>
Date: Thu, 23 May 2024 18:20:16 +0000
Subject: [PATCH] Update .gitlab-ci.yml

---
 .gitlab-ci.yml | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 03c012d..e6084b1 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -18,6 +18,9 @@ before_script:
 
 train:
   stage: train
+   tags:
+    - pages
+
   script:
     - source venv/bin/activate
     - python train_model.py
@@ -26,14 +29,11 @@ train:
       - trained_model.joblib
       - tokenizer.joblib
 
-test:
-  stage: test
-  script:
-    - source venv/bin/activate
-    - python test_model.py
-
 deploy:
   stage: deploy
+   tags:
+    - pages
+
   script:
     - echo "Deploying model locally..."
     - source venv/bin/activate
-- 
GitLab