test: Upgrade pytorch_lightning to fix invalid metadata

project-codeflare · Jun 28, 2024 · fa186c2 · fa186c2
1 parent 90ba84a
commit fa186c2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
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