Set tfjob defaults in test utils (#1071)

kubeflow · Aug 19, 2019 · d164c36 · d164c36
1 parent d2499dc
commit d164c36
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/pkg/common/util/v1/testutil/tfjob.go b/pkg/common/util/v1/testutil/tfjob.go
@@ -115,6 +115,7 @@ func NewTFJob(worker, ps int) *tfv1.TFJob {
 			TFReplicaSpecs: make(map[tfv1.TFReplicaType]*common.ReplicaSpec),
 		},
 	}
+	tfv1.SetObjectDefaults_TFJob(tfJob)
 
 	if worker > 0 {
 		worker := int32(worker)

diff --git a/pkg/common/util/v1beta2/testutil/tfjob.go b/pkg/common/util/v1beta2/testutil/tfjob.go
@@ -115,6 +115,7 @@ func NewTFJob(worker, ps int) *tfv1beta2.TFJob {
 			TFReplicaSpecs: make(map[tfv1beta2.TFReplicaType]*common.ReplicaSpec),
 		},
 	}
+	tfv1beta2.SetObjectDefaults_TFJob(tfJob)
 
 	if worker > 0 {
 		worker := int32(worker)