nit: try to simplify e2e main.go (#359)
nit: try to simplify e2e main.go (#359)

I'm trying to simplify main.go files but this would better be taking baby steps.

this very first step is to try to clean up spec.
kubeflow/training-operator,kubeflow/training-operator,kubeflow/training-operator