Name label (#105)

Add Name of the job to the labels.

This will make it easier to fetch resources belonging to a specific job.

* Fixes #72.

pkg/trainer/replicas.go

@@ -94,7 +94,8 @@ func (s *TFReplicaSet) Labels() KubernetesLabels {
 		"job_type":  string(s.Spec.TfReplicaType),
 		// runtime_id is set by Job.setup, which is called after the TfReplicaSet is created.
 		// this is why labels aren't a member variable.
-		"runtime_id": s.Job.job.Spec.RuntimeId})
+		"runtime_id": s.Job.job.Spec.RuntimeId,
+		"tf_job_name": s.Job.job.Metadata.Name,})
 }
 
 // Transforms the tfconfig to work with grpc_tensorflow_server

pkg/trainer/replicas_test.go

@@ -22,6 +22,9 @@ func TestTFReplicaSet(t *testing.T) {
 	clientSet := fake.NewSimpleClientset()
 
 	jobSpec := &spec.TfJob{
+		Metadata: meta_v1.ObjectMeta {
+			Name: "some-job",
+		},
 		Spec: spec.TfJobSpec{
 			RuntimeId: "some-runtime",
 			ReplicaSpecs: []*spec.TfReplicaSpec{
@@ -69,6 +72,7 @@ func TestTFReplicaSet(t *testing.T) {
 			"task_index": fmt.Sprintf("%v", index),
 			"job_type":   "PS",
 			"runtime_id": "some-runtime",
+			"tf_job_name": "some-job",
 		}
 
 		// Check that a service was created.

pkg/trainer/tensorboard.go

@@ -173,6 +173,7 @@ func (s *TBReplicaSet) Labels() KubernetesLabels {
 		"tensorflow.org":  "",
 		"runtime_id": s.Job.job.Spec.RuntimeId,
 		"app":        "tensorboard",
+		"tf_job_name": s.Job.job.Metadata.Name,
 	})
 }
 

pkg/trainer/tensorboard_test.go

@@ -19,6 +19,9 @@ func TestTBReplicaSet(t *testing.T) {
 	clientSet := fake.NewSimpleClientset()
 
 	jobSpec := &spec.TfJob{
+		Metadata: meta_v1.ObjectMeta {
+			Name: "some-job",
+		},
 		Spec: spec.TfJobSpec{
 			RuntimeId: "some-runtime",
 			ReplicaSpecs: []*spec.TfReplicaSpec{
@@ -59,6 +62,7 @@ func TestTBReplicaSet(t *testing.T) {
 		"tensorflow.org":  "",
 		"app":        "tensorboard",
 		"runtime_id": "some-runtime",
+		"tf_job_name": "some-job",
 	}
 
 	// Check that a service was created.