fix import error

GRAAL-Research · Nov 14, 2023 · 03f4629 · 03f4629
1 parent e6b29cb
commit 03f4629
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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 → src/training/evaluate_metrics.py b/src/training/evaluate.py → src/training/evaluate_metrics.py
diff --git 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,
 )