Fix the annotation missing issue for ray workload (#218)

vllm-project · Sep 25, 2024 · aa3ed95 · aa3ed95
1 parent d5b014d
commit aa3ed95
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/controller/rayclusterfleet/rayclusterfleet_controller.go b/pkg/controller/rayclusterfleet/rayclusterfleet_controller.go
@@ -246,7 +246,7 @@ func (r *RayClusterFleetReconciler) createNewReplicaSet(ctx context.Context, d *
 			GenerateName: d.Name + "-",
 			Namespace:    d.Namespace,
 			Labels:       d.Spec.Template.Labels,
-			Annotations:  make(map[string]string),
+			Annotations:  d.Spec.Template.Annotations,
 			OwnerReferences: []metav1.OwnerReference{
 				*metav1.NewControllerRef(d, controllerKind),
 			},

diff --git a/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_utils.go b/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_utils.go
@@ -83,7 +83,7 @@ func constructRayCluster(replicaset *orchestrationv1alpha1.RayClusterReplicaSet)
 			GenerateName: replicaset.Name + "-",
 			Namespace:    replicaset.Namespace,
 			Labels:       replicaset.Spec.Template.Labels,
-			Annotations:  make(map[string]string),
+			Annotations:  replicaset.Spec.Template.Annotations,
 			OwnerReferences: []metav1.OwnerReference{
 				*metav1.NewControllerRef(replicaset, controllerKind),
 			},