From ffe6a302a70c6dade9200e2c678a30c30756a6f8 Mon Sep 17 00:00:00 2001
From: Paul Andrey <paul.andrey@inria.fr>
Date: Fri, 15 Mar 2024 13:58:21 +0100
Subject: [PATCH] Fix Torch 1.13 unit tests launch.
---
scripts/run_tests.sh | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/scripts/run_tests.sh b/scripts/run_tests.sh
index 605532c2..b2cb7703 100644
--- a/scripts/run_tests.sh
+++ b/scripts/run_tests.sh
@@ -203,7 +203,7 @@ run_torch13_tests() {
'
echo "Installing torch 1.13 and its co-dependencies."
TORCH_DEPS=$(pip freeze | grep -e torch -e opacus)
- pip install .[torch1]
+ pip install "opacus == 1.4.0" "torch ~=1.13.0"
if [[ $? -eq 0 ]]; then
echo "Running unit tests for torch 1.13."
command="pytest $@
@@ -211,6 +211,7 @@ run_torch13_tests() {
test/model/test_torch_model.py
"
echo -e "\e[34m$command\e[0m"
+ $command
status=$?
else
echo "\e[31mSkipping tests as installation failed.\e[0m"
--
GitLab