chore(components): Move vertex get model components

PiperOrigin-RevId: 581081482
kubeflow · Nov 10, 2023 · e3f0275 · e3f0275
1 parent a927984
commit e3f0275
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/components/google-cloud/RELEASE.md b/components/google-cloud/RELEASE.md
@@ -4,6 +4,8 @@
 * Fix the output parameter `output_dir` of `preview.automl.vision.DataConverterJobOp`.
 * Fix batch prediction model parameters payload sanitization error .
 * Group `preview.llm.rlhf_pipeline` components for better readability.
+* Group `preview.llm.rlhf_pipeline` components for better readability.
+* Add Vertex model get method.
 
 ## Release 2.5.0
 * Upload tensorboard metrics from `preview.llm.rlhf_pipeline` if a `tensorboard_resource_id` is provided at runtime.

diff --git a/components/google-cloud/google_cloud_pipeline_components/v1/model/__init__.py b/components/google-cloud/google_cloud_pipeline_components/v1/model/__init__.py
@@ -17,10 +17,12 @@
 
 from google_cloud_pipeline_components.v1.model.delete_model.component import model_delete as ModelDeleteOp
 from google_cloud_pipeline_components.v1.model.export_model.component import model_export as ModelExportOp
+from google_cloud_pipeline_components.v1.model.get_model.component import get_vertex_model as ModelGetOp
 from google_cloud_pipeline_components.v1.model.upload_model.component import model_upload as ModelUploadOp
 
 __all__ = [
     'ModelExportOp',
     'ModelUploadOp',
     'ModelDeleteOp',
+    'ModelGetOp',
 ]
diff --git a/...mplementation/model/get_model/__init__.py → ...components/v1/model/get_model/__init__.py b/...mplementation/model/get_model/__init__.py → ...components/v1/model/get_model/__init__.py
diff --git a/...plementation/model/get_model/component.py → ...omponents/v1/model/get_model/component.py b/...plementation/model/get_model/component.py → ...omponents/v1/model/get_model/component.py