test: Upgrade Torch dependencies

astefanutti · astefanutti · commit 75ae18eaab72 · 2024-06-28T14:48:07.000+02:00
diff --git a/test/e2e/mnist_pip_requirements.txt b/test/e2e/mnist_pip_requirements.txt
@@ -1,3 +1,3 @@
 pytorch_lightning==1.9.5
-torchmetrics==0.9.1
-torchvision==0.12.0
+torchmetrics==1.4.0.post0
+torchvision==0.18.1
diff --git a/test/e2e/mnist_rayjob_raycluster_test.go b/test/e2e/mnist_rayjob_raycluster_test.go
@@ -314,8 +314,8 @@ func constructRayJob(_ Test, namespace *corev1.Namespace, rayCluster *rayv1.RayC
 			RuntimeEnvYAML: `
   pip:
     - pytorch_lightning==1.9.5
-    - torchmetrics==0.9.1
-    - torchvision==0.12.0
+    - torchmetrics==1.4.0.post0
+	- torchvision==0.18.1
   env_vars:
     MNIST_DATASET_URL: "` + GetMnistDatasetURL() + `"
     PIP_INDEX_URL: "` + GetPipIndexURL() + `"
diff --git a/test/odh/resources/requirements.txt b/test/odh/resources/requirements.txt
@@ -2,5 +2,5 @@
 {{.PipTrustedHost}}
 pytorch_lightning==1.9.5
 ray_lightning
-torchmetrics==0.9.1
-torchvision==0.12.0
+torchmetrics==1.4.0.post0
+torchvision==0.18.1
