Override RayJob configs

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Jul 26, 2023 · 6c9f374 · 6c9f374
1 parent 3405f89
commit 6c9f374
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/go/tasks/plugins/k8s/ray/ray.go b/go/tasks/plugins/k8s/ray/ray.go
@@ -49,6 +49,7 @@ func (rayJobResourceHandler) BuildResource(ctx context.Context, taskCtx pluginsC
 		return nil, flyteerr.Errorf(flyteerr.BadTaskSpecification, "nil task specification")
 	}
 
+	config := taskTemplate.GetConfig()
 	rayJob := plugins.RayJob{}
 	err = utils.UnmarshalStruct(taskTemplate.GetCustom(), &rayJob)
 	if err != nil {
@@ -149,6 +150,7 @@ func (rayJobResourceHandler) BuildResource(ctx context.Context, taskCtx pluginsC
 		RuntimeEnv:               rayJob.RuntimeEnv,
 	}
 
+	objectMeta.Namespace = config["namespace"]
 	rayJobObject := rayv1alpha1.RayJob{
 		TypeMeta: metav1.TypeMeta{
 			Kind:       KindRayJob,