add kubeconfig flag

Signed-off-by: yeya24 <yb532204897@gmail.com>
kubeflow · Jul 30, 2019 · 284145a · 284145a
1 parent 4028276
commit 284145a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/cmd/pytorch-operator.v1/app/options/options.go b/cmd/pytorch-operator.v1/app/options/options.go
@@ -44,6 +44,8 @@ func NewServerOption() *ServerOption {
 
 // AddFlags adds flags for a specific CMServer to the specified FlagSet.
 func (s *ServerOption) AddFlags(fs *flag.FlagSet) {
+	fs.StringVar(&s.Kubeconfig, "kubeconfig", "", "The path of kubeconfig file")
+
 	fs.StringVar(&s.MasterURL, "master", "",
 		`The url of the Kubernetes API server,
 		 will overrides any value in kubeconfig, only required if out-of-cluster.`)

diff --git a/cmd/pytorch-operator.v1beta2/app/options/options.go b/cmd/pytorch-operator.v1beta2/app/options/options.go
@@ -39,6 +39,8 @@ func NewServerOption() *ServerOption {
 
 // AddFlags adds flags for a specific CMServer to the specified FlagSet.
 func (s *ServerOption) AddFlags(fs *flag.FlagSet) {
+	fs.StringVar(&s.Kubeconfig, "kubeconfig", "", "The path of kubeconfig file")
+
 	fs.StringVar(&s.MasterURL, "master", "",
 		`The url of the Kubernetes API server,
 		 will overrides any value in kubeconfig, only required if out-of-cluster.`)