diff --git a/src/README.md b/src/README.md
index 7245c17..95f7035 100644
--- a/src/README.md
+++ b/src/README.md
@@ -6,7 +6,7 @@ our [article](https://www.frontiersin.org/articles/10.3389/frai.2023.1223924/ful
 ## To Reproduce our Article Results
 
 The `src` directory is public to make our results more reproducible. One can reproduce our results by
-using the codebase.
+using the codebase. It was coded in Python 3.11.
 
 Note that this codebase is different from the one used for our article. In our article, we create ten different
 train/dev/test splits using a different seed each time, and we also create, on run-time, data augmentation generation
diff --git a/src/requirements.txt b/src/requirements.txt
index cc45fa3..11dfb01 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -13,6 +13,9 @@ wandb
 scikit-learn
 accelerate
 python2latex
+poutyne
+torchmetrics
+torch>=1.6.0,<2
 git+https://github.com/google-research/bleurt.git
 
 # Install a modified version of the LENS codebase by fixing PyPi broken build
diff --git a/src/training/evaluate.py b/src/training/evaluate_metrics.py
similarity index 100%
rename from src/training/evaluate.py
rename to src/training/evaluate_metrics.py
diff --git a/src/training/few_shot_training.py b/src/training/few_shot_training.py
index bd276b3..8e32e60 100644
--- a/src/training/few_shot_training.py
+++ b/src/training/few_shot_training.py
@@ -13,7 +13,7 @@
 )
 
 from metrics.metrics import compute_metrics, eval_compute_metrics_identical, eval_compute_metrics_unrelated
-from src.training.evaluate import compute_other_metrics_performance
+from evaluate_metrics import compute_other_metrics_performance
 from tools import (
     bool_parse,
 )