Skip to content

Commit c2a0b78

Browse files
committed
test: Upgrade pytorch_lightning to fix invalid metadata
1 parent 6e66c51 commit c2a0b78

File tree

3 files changed

+4
-14
lines changed

3 files changed

+4
-14
lines changed

test/e2e/mnist_pip_requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
pytorch_lightning==1.5.10
1+
pytorch_lightning==1.9.5
22
torchmetrics==0.9.1
33
torchvision==0.12.0

test/e2e/mnist_rayjob_raycluster_test.go

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ func constructRayJob(_ Test, namespace *corev1.Namespace, rayCluster *rayv1.RayC
313313
Entrypoint: "python /home/ray/jobs/mnist.py",
314314
RuntimeEnvYAML: `
315315
pip:
316-
- pytorch_lightning==1.5.10
316+
- pytorch_lightning==1.9.5
317317
- torchmetrics==0.9.1
318318
- torchvision==0.12.0
319319
env_vars:
@@ -325,17 +325,7 @@ func constructRayJob(_ Test, namespace *corev1.Namespace, rayCluster *rayv1.RayC
325325
RayJobDefaultClusterSelectorKey: rayCluster.Name,
326326
},
327327
ShutdownAfterJobFinishes: false,
328-
SubmitterPodTemplate: &corev1.PodTemplateSpec{
329-
Spec: corev1.PodSpec{
330-
RestartPolicy: corev1.RestartPolicyNever,
331-
Containers: []corev1.Container{
332-
{
333-
Image: GetRayImage(),
334-
Name: "rayjob-submitter-pod",
335-
},
336-
},
337-
},
338-
},
328+
SubmissionMode: rayv1.HTTPMode,
339329
},
340330
}
341331
}

test/odh/resources/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
{{.PipIndexUrl}}
22
{{.PipTrustedHost}}
3-
pytorch_lightning==1.5.10
3+
pytorch_lightning==1.9.5
44
ray_lightning
55
torchmetrics==0.9.1
66
torchvision==0.12.0

0 commit comments

Comments
 (0)