add kubeconfig flag (#192)

Signed-off-by: yeya24 <yb532204897@gmail.com>
kubeflow · Jul 31, 2019 · e98eb3c · e98eb3c
1 parent 91ec254
commit e98eb3c
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.`)