in pkg/util/testingjobs/rayjob/wrappers.go [33:77]
func MakeJob(name, ns string) *JobWrapper {
return &JobWrapper{rayjobapi.RayJob{
ObjectMeta: metav1.ObjectMeta{
Name: name,
Namespace: ns,
Annotations: make(map[string]string, 1),
},
Spec: rayjobapi.RayJobSpec{
ShutdownAfterJobFinishes: true,
RayClusterSpec: &rayjobapi.RayClusterSpec{
HeadGroupSpec: rayjobapi.HeadGroupSpec{
RayStartParams: map[string]string{"p1": "v1"},
Template: corev1.PodTemplateSpec{
Spec: corev1.PodSpec{
Containers: []corev1.Container{
{
Name: "head-container",
},
},
},
},
},
WorkerGroupSpecs: []rayjobapi.WorkerGroupSpec{
{
GroupName: "workers-group-0",
Replicas: pointer.Int32(1),
MinReplicas: pointer.Int32(0),
MaxReplicas: pointer.Int32(10),
RayStartParams: map[string]string{"p1": "v1"},
Template: corev1.PodTemplateSpec{
Spec: corev1.PodSpec{
Containers: []corev1.Container{
{
Name: "worker-container",
},
},
},
},
},
},
},
Suspend: true,
},
}}
}