Handle errors of adding tfjob scheme

To fix the gometalinter error
kubeflow · Aug 19, 2019 · ddf4d85 · ddf4d85
1 parent d0c924c
commit ddf4d85
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/pkg/controller.v1/tensorflow/controller.go b/pkg/controller.v1/tensorflow/controller.go
@@ -118,7 +118,10 @@ func NewTFController(
 	tfJobInformerFactory tfjobinformers.SharedInformerFactory,
 	option options.ServerOption) *TFController {
 
-	tfjobscheme.AddToScheme(scheme.Scheme)
+	err := tfjobscheme.AddToScheme(scheme.Scheme)
+	if err != nil {
+		log.Fatalf("Failed to add tfjob scheme: %v", err)
+	}
 
 	log.Info("Creating TFJob controller")
 	// Create new TFController.

diff --git a/pkg/controller.v1beta2/tensorflow/controller.go b/pkg/controller.v1beta2/tensorflow/controller.go
@@ -111,7 +111,10 @@ func NewTFController(
 	tfJobInformerFactory tfjobinformers.SharedInformerFactory,
 	option options.ServerOption) *TFController {
 
-	tfjobscheme.AddToScheme(scheme.Scheme)
+	err := tfjobscheme.AddToScheme(scheme.Scheme)
+	if err != nil {
+		log.Fatalf("Failed to add tfjob scheme: %v", err)
+	}
 
 	log.Info("Creating TFJob controller")
 	// Create new TFController.