chore(components/google-cloud): Alias create_custom_training_job_op_f…

…rom_component to create_custom_training_job_from_component PiperOrigin-RevId: 441899934
kubeflow · Apr 15, 2022 · a85dc4f · a85dc4f
1 parent d452261
commit a85dc4f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/components/google-cloud/google_cloud_pipeline_components/v1/custom_job/__init__.py b/components/google-cloud/google_cloud_pipeline_components/v1/custom_job/__init__.py
@@ -15,6 +15,10 @@
 
 import os
 
+from .utils import create_custom_training_job_op_from_component
+# Aliasing for better readability
+create_custom_training_job_from_component = create_custom_training_job_op_from_component
+
 try:
   from kfp.v2.components import load_component_from_file
 except ImportError:
@@ -23,9 +27,8 @@
 __all__ = [
     'CustomTrainingJobOp',
     'create_custom_training_job_op_from_component',
+    'create_custom_training_job_from_component',
 ]
 
-
 CustomTrainingJobOp = load_component_from_file(
     os.path.join(os.path.dirname(__file__), 'component.yaml'))
-
diff --git a/components/google-cloud/google_cloud_pipeline_components/v1/custom_job/utils.py b/components/google-cloud/google_cloud_pipeline_components/v1/custom_job/utils.py
@@ -32,6 +32,8 @@
 _EXECUTOR_PLACE_HOLDER_REPLACEMENT = '{{$.json_escape[1]}}'
 
 
+# This method is aliased to "create_custom_training_job_from_component" for
+# better readability
 def create_custom_training_job_op_from_component(
     component_spec: Callable,  # pylint: disable=g-bare-generic
     display_name: Optional[str] = '',