Mentions légales du service

Skip to content
Snippets Groups Projects
Verified Commit ffe6a302 authored by ANDREY Paul's avatar ANDREY Paul
Browse files

Fix Torch 1.13 unit tests launch.

parent aff0ff31
No related branches found
No related tags found
1 merge request!63Finalize DecLearn v2.4.0.
Pipeline #943513 passed
Stage: clean
Stage: test
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 1 deletion
scripts/run_tests.sh
+ 2
− 1
View file @ ffe6a302
...@@ -203,7 +203,7 @@ run_torch13_tests() { ...@@ -203,7 +203,7 @@ run_torch13_tests() {
' '
echo "Installing torch 1.13 and its co-dependencies." echo "Installing torch 1.13 and its co-dependencies."
TORCH_DEPS=$(pip freeze | grep -e torch -e opacus) 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 if [[ $? -eq 0 ]]; then
echo "Running unit tests for torch 1.13." echo "Running unit tests for torch 1.13."
command="pytest $@ command="pytest $@
...@@ -211,6 +211,7 @@ run_torch13_tests() { ...@@ -211,6 +211,7 @@ run_torch13_tests() {
test/model/test_torch_model.py test/model/test_torch_model.py
" "
echo -e "\e[34m$command\e[0m" echo -e "\e[34m$command\e[0m"
$command
status=$? status=$?
else else
echo "\e[31mSkipping tests as installation failed.\e[0m" echo "\e[31mSkipping tests as installation failed.\e[0m"
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment