Set scheduler restart policy to Always (#367)

Signed-off-by: Bernhard Stadlbauer <[email protected]>
flyteorg · Jun 29, 2023 · f7b37aa · f7b37aa
1 parent df62599
commit f7b37aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/go/tasks/plugins/k8s/dask/dask.go b/go/tasks/plugins/k8s/dask/dask.go
@@ -237,7 +237,7 @@ func createSchedulerSpec(cluster plugins.DaskScheduler, clusterName string, defa
 
 	return &daskAPI.SchedulerSpec{
 		Spec: v1.PodSpec{
-			RestartPolicy: v1.RestartPolicyNever,
+			RestartPolicy: v1.RestartPolicyAlways,
 			Containers: []v1.Container{
 				{
 					Name:      "scheduler",

diff --git a/go/tasks/plugins/k8s/dask/dask_test.go b/go/tasks/plugins/k8s/dask/dask_test.go
@@ -228,7 +228,7 @@ func TestBuildResourceDaskHappyPath(t *testing.T) {
 			Protocol:      "TCP",
 		},
 	}
-	assert.Equal(t, v1.RestartPolicyNever, schedulerSpec.RestartPolicy)
+	assert.Equal(t, v1.RestartPolicyAlways, schedulerSpec.RestartPolicy)
 	assert.Equal(t, defaultTestImage, schedulerSpec.Containers[0].Image)
 	assert.Equal(t, defaultResources, schedulerSpec.Containers[0].Resources)
 	assert.Equal(t, []string{"dask-scheduler"}, schedulerSpec.Containers[0].Args)