From 9d9fee440e529be4239820b658dc915bc0364c61 Mon Sep 17 00:00:00 2001
From: terrytangyuan <terrytangyuan@gmail.com>
Date: Wed, 27 May 2020 10:39:18 -0400
Subject: [PATCH] Fix defaults_test.go

Signed-off-by: terrytangyuan <terrytangyuan@gmail.com>
---
 pkg/apis/tensorflow/v1/defaults_test.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkg/apis/tensorflow/v1/defaults_test.go b/pkg/apis/tensorflow/v1/defaults_test.go
index d1a3b51323..39a8c2c51d 100644
--- a/pkg/apis/tensorflow/v1/defaults_test.go
+++ b/pkg/apis/tensorflow/v1/defaults_test.go
@@ -51,8 +51,11 @@ func expectedTFJob(cleanPodPolicy common.CleanPodPolicy, restartPolicy common.Re
 		)
 	}
 
+	defaultSuccessPolicy := SuccessPolicyDefault
+
 	return &TFJob{
 		Spec: TFJobSpec{
+			SuccessPolicy:  &defaultSuccessPolicy,
 			CleanPodPolicy: &cleanPodPolicy,
 			TFReplicaSpecs: map[TFReplicaType]*common.ReplicaSpec{
 				TFReplicaTypeWorker: &common.ReplicaSpec{