Troubleshooting -- trying to find which tests in the "integration_tes…

…ts_e" group are failing.
ludwig-ai · Oct 18, 2023 · d0dab00 · d0dab00
1 parent 23aab9b
commit d0dab00
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -6,10 +6,17 @@ scipy>=0.18
 tabulate>=0.7
 scikit-learn
 tqdm
-torch>=1.13.0
+# TODO: <Alex>ALEX</Alex>
+torch>=1.13.0,<=2.0.1
+# TODO: <Alex>ALEX</Alex>
 torchaudio
 torchtext
-torchvision
+# TODO: <Alex>ALEX</Alex>
+#torchvision
+# TODO: <Alex>ALEX</Alex>
+# TODO: <Alex>ALEX</Alex>
+torchvision<0.16
+# TODO: <Alex>ALEX</Alex>
 pydantic<2.0
 transformers>=4.33.2
 tokenizers>=0.13.3