[Add] Propagate labels from NB to StatefulSets

This commit propagates labels from Notebooks to statefulsets.
This is required to enable NBs to be managed by Kueue. When a user
adds the label to point a NB to the right LocalQueue where it
needs to be admitted, it needs to propagate to the underlying
StatefulSet, so that Kueue can manage it.

Signed-off-by: Varsha Prasad Narsing <[email protected]>
(cherry picked from commit f6abf13)

Author: varshaprasad96

components/notebook-controller/controllers/notebook_controller.go

@@ -439,6 +439,7 @@ func generateStatefulSet(instance *v1beta1.Notebook, isGenerateName bool) *appsv
 	ssObjectMeta := metav1.ObjectMeta{
 		Name:      instance.Name,
 		Namespace: instance.Namespace,
+		Labels:    map[string]string{},
 	}
 	if isGenerateName {
 		ssObjectMeta = metav1.ObjectMeta{
@@ -470,6 +471,11 @@ func generateStatefulSet(instance *v1beta1.Notebook, isGenerateName bool) *appsv
 		},
 	}
 
+	sl := &ss.Labels
+	for k, v := range instance.Labels {
+		(*sl)[k] = v
+	}
+
 	// copy all of the Notebook labels to the pod including poddefault related labels
 	l := &ss.Spec.Template.ObjectMeta.Labels
 	for k, v := range instance.ObjectMeta.Labels {