fix: Fix test cases

Signed-off-by: Ce Gao <ce.gao@outlook.com>
kubeflow · Nov 30, 2021 · 892a85d · 892a85d
1 parent 95999cf
commit 892a85d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/controller.v1/pytorch/pytorchjob_controller_suite_test.go b/pkg/controller.v1/pytorch/pytorchjob_controller_suite_test.go
@@ -20,6 +20,7 @@ import (
 	"testing"
 
 	v1 "github.com/kubeflow/training-operator/pkg/apis/pytorch/v1"
+	"github.com/kubeflow/training-operator/pkg/config"
 
 	. "github.com/onsi/ginkgo"
 	"github.com/onsi/gomega"
@@ -70,6 +71,10 @@ var _ = BeforeSuite(func() {
 	err = v1.AddToScheme(scheme.Scheme)
 	Expect(err).NotTo(HaveOccurred())
 
+	// Set default config.
+	config.Config.PyTorchInitContainerImage = config.PyTorchInitContainerImageDefault
+	config.Config.PyTorchInitContainerTemplateFile = config.PyTorchInitContainerTemplateFileDefault
+
 	//+kubebuilder:scaffold:scheme
 
 	testK8sClient, err = client.New(cfg, client.Options{Scheme: scheme.Scheme})