diff --git a/requirements_tests.txt b/requirements_tests.txt index ad0efd83..a3bd7c7b 100644 --- a/requirements_tests.txt +++ b/requirements_tests.txt @@ -2,6 +2,7 @@ pytest testbook nbformat + torch torchvision galois