From c4cc8761b3399644fbae799679f3c1971c38401b Mon Sep 17 00:00:00 2001 From: Kevin Su Date: Wed, 12 Aug 2020 18:05:58 +0800 Subject: [PATCH] Rename func --- .../plugins/k8s/kfoperators/tensorflow/tensorflow_test.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/go/tasks/plugins/k8s/kfoperators/tensorflow/tensorflow_test.go b/go/tasks/plugins/k8s/kfoperators/tensorflow/tensorflow_test.go index a013f60cd..8b8e73d63 100644 --- a/go/tasks/plugins/k8s/kfoperators/tensorflow/tensorflow_test.go +++ b/go/tasks/plugins/k8s/kfoperators/tensorflow/tensorflow_test.go @@ -70,7 +70,7 @@ func dummyTensorflowCustomObj(workers int32, psReplicas int32, chiefReplicas int } } -func dummySparkTaskTemplate(id string, tensorflowCustomObj *plugins.DistributedTensorflowTrainingTask) *core.TaskTemplate { +func dummyTensorflowTaskTemplate(id string, tensorflowCustomObj *plugins.DistributedTensorflowTrainingTask) *core.TaskTemplate { tfObjJSON, err := utils.MarshalToString(tensorflowCustomObj) if err != nil { @@ -247,7 +247,7 @@ func dummyTensorflowJobResource(tensorflowResourceHandler tensorflowOperatorReso } tfObj := dummyTensorflowCustomObj(workers, psReplicas, chiefReplicas) - taskTemplate := dummySparkTaskTemplate("the job", tfObj) + taskTemplate := dummyTensorflowTaskTemplate("the job", tfObj) resource, err := tensorflowResourceHandler.BuildResource(context.TODO(), dummyTensorflowTaskContext(taskTemplate)) if err != nil { panic(err) @@ -273,7 +273,7 @@ func TestBuildResourceTensorflow(t *testing.T) { tensorflowResourceHandler := tensorflowOperatorResourceHandler{} tfObj := dummyTensorflowCustomObj(100, 1,1) - taskTemplate := dummySparkTaskTemplate("the job", tfObj) + taskTemplate := dummyTensorflowTaskTemplate("the job", tfObj) resource, err := tensorflowResourceHandler.BuildResource(context.TODO(), dummyTensorflowTaskContext(taskTemplate)) assert.NoError(t, err)