From a34fe44254dfdd018e999011c28afb9b3dc3db0f Mon Sep 17 00:00:00 2001
From: s424714 <filpat@st.amu.edu.pl>
Date: Wed, 10 May 2023 16:45:36 +0200
Subject: [PATCH] feat: update build_id argument

---
 Jenkinsfile-eval | 2 +-
 src/main.py      | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/Jenkinsfile-eval b/Jenkinsfile-eval
index fddd348..3cbfce0 100644
--- a/Jenkinsfile-eval
+++ b/Jenkinsfile-eval
@@ -67,7 +67,7 @@ node {
                 {
                     sh 'mkdir -p ./.cache'
                     // sh ""
-                    sh 'python ./src/main.py --test '
+                    sh 'python ./src/main.py --test --build_id=${currentBuild.number} '
                     sh 'python ./src/plot.py '
                     sh "cp ./results/*.csv ${WORKSPACE}" 
                     sh "cp ./results/*.png ${WORKSPACE}" 
diff --git a/src/main.py b/src/main.py
index 07037c6..050fa30 100644
--- a/src/main.py
+++ b/src/main.py
@@ -21,6 +21,7 @@ parser.add_argument("--test", action="store_true", default=False)
 parser.add_argument("--model_path", type=str, default="results/model.pt")
 parser.add_argument("--results_path", type=str, default="results/results.csv")
 parser.add_argument("--data_acc_path", type=str, default="./results/acc.csv")
+parser.add_argument("--build_id", type=int, default=0)
 
 # HYPER PARAMETERS
 parser.add_argument("--batch", "-b", type=int, default=2)
@@ -83,4 +84,8 @@ if __name__ == "__main__":
             batch_size=BATCH_SIZE,
         )
         utils.save_results(labels=test_data["label"], results=results, file_path=args.results_path)
-        utils.save_data_to_csv(args.data_acc_path, accuracy)
+        utils.save_data_to_csv(
+            file_path=args.data_acc_path,
+            build_id=args.build_id,
+            data=accuracy,
+        )