fix: Refactor option imports flow

Signed-off-by: Abhijeet Dhumal <[email protected]>

Author: abhijeet-dhumal

kubeflow/trainer/backends/kubernetes/backend_test.py

@@ -35,7 +35,7 @@
 from kubeflow.trainer.backends.kubernetes.backend import KubernetesBackend
 import kubeflow.trainer.backends.kubernetes.utils as utils
 from kubeflow.trainer.constants import constants
-from kubeflow.trainer.options.kubernetes import (
+from kubeflow.trainer.options import (
     Annotations,
     Labels,
     SpecAnnotations,

kubeflow/trainer/backends/localprocess/backend_test.py

@@ -26,10 +26,10 @@
     LocalRuntimeTrainer,
 )
 from kubeflow.trainer.constants import constants
-from kubeflow.trainer.options import Name
-from kubeflow.trainer.options.kubernetes import (
+from kubeflow.trainer.options import (
     Annotations,
     Labels,
+    Name,
     PodTemplateOverride,
     PodTemplateOverrides,
 )

kubeflow/trainer/options/__init__.py

@@ -12,8 +12,41 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Common training options for the Kubeflow Trainer SDK."""
+"""Training options for the Kubeflow Trainer SDK.
+
+All options are available from this single import location:
+    from kubeflow.trainer.options import Name, Labels, PodTemplateOverrides, ...
+
+Options self-validate their backend compatibility at runtime.
+Check each option's docstring for supported backends.
+"""
 
 from kubeflow.trainer.options.common import Name
+from kubeflow.trainer.options.kubernetes import (
+    Annotations,
+    ContainerOverride,
+    Labels,
+    PodSpecOverride,
+    PodTemplateOverride,
+    PodTemplateOverrides,
+    SpecAnnotations,
+    SpecLabels,
+    TrainerArgs,
+    TrainerCommand,
+)
 
-__all__ = ["Name"]
+__all__ = [
+    # Common options (all backends)
+    "Name",
+    # Kubernetes options
+    "Annotations",
+    "ContainerOverride",
+    "Labels",
+    "PodSpecOverride",
+    "PodTemplateOverride",
+    "PodTemplateOverrides",
+    "SpecAnnotations",
+    "SpecLabels",
+    "TrainerArgs",
+    "TrainerCommand",
+]

kubeflow/trainer/options/kubernetes_test.py

@@ -18,11 +18,11 @@
 
 from kubeflow.trainer.backends.kubernetes.backend import KubernetesBackend
 from kubeflow.trainer.backends.localprocess.backend import LocalProcessBackend
-from kubeflow.trainer.options import Name
-from kubeflow.trainer.options.kubernetes import (
+from kubeflow.trainer.options import (
     Annotations,
     ContainerOverride,
     Labels,
+    Name,
     PodTemplateOverride,
     PodTemplateOverrides,
     SpecAnnotations,