test: Upgrade pytorch_lightning to fix invalid metadata

astefanutti · astefanutti · commit fa186c257f07 · 2024-06-28T17:08:55.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.5.10
+pytorch_lightning==1.9.5
 torchmetrics==0.9.1
 torchvision==0.12.0
diff --git a/test/e2e/mnist_rayjob_raycluster_test.go b/test/e2e/mnist_rayjob_raycluster_test.go
@@ -313,7 +313,7 @@ func constructRayJob(_ Test, namespace *corev1.Namespace, rayCluster *rayv1.RayC
 			Entrypoint: "python /home/ray/jobs/mnist.py",
 			RuntimeEnvYAML: `
   pip:
-    - pytorch_lightning==1.5.10
+    - pytorch_lightning==1.9.5
     - torchmetrics==0.9.1
     - torchvision==0.12.0
   env_vars:
diff --git a/test/odh/resources/requirements.txt b/test/odh/resources/requirements.txt
@@ -1,6 +1,6 @@
 {{.PipIndexUrl}}
 {{.PipTrustedHost}}
-pytorch_lightning==1.5.10
+pytorch_lightning==1.9.5
 ray_lightning
 torchmetrics==0.9.1
 torchvision==0.12.0
