Skip to content

Commit

Permalink
chore(components/google-cloud): Add ModelExport component (#5554)
Browse files Browse the repository at this point in the history
* Add ModelExportOp

* Lint pipeline components

* Point to git for google-cloud-aiplatform instead of PyPi

* Fix dependencies.py to be setuptools compliant

* Update dependency to use https instead of ssh

* Update lint with yapf
  • Loading branch information
vinnysenthil authored Jun 10, 2021
1 parent ea5c697 commit 538e4ef
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 11 deletions.
22 changes: 11 additions & 11 deletions components/google-cloud/dependencies.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,21 +13,21 @@
# limitations under the License.
"""Setup configuration of Google Cloud Pipeline Components client side libraries."""


def make_required_install_packages():
return [
"kfp>=1.4.0,<2.0.0",
"google-cloud-aiplatform>=1.0.1",
]
return [
"kfp>=1.4.0,<2.0.0",
"google-cloud-aiplatform>=1.0.1",
]


def make_required_test_packages():
return make_required_install_packages() + [
"mock>=4.0.0",
"flake8>=3.0.0",
"pytest>=6.0.0",
]
return make_required_install_packages() + [
"mock>=4.0.0",
"flake8>=3.0.0",
"pytest>=6.0.0",
]


def make_dependency_links():
# TODO: change to google-cloud-aiplatform
return []
return []
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,10 @@
aiplatform_sdk.Model, aiplatform_sdk.Model.upload
)

ModelExportOp = utils.convert_method_to_component(
aiplatform_sdk.Model, aiplatform_sdk.Model.export_model
)

EndpointCreateOp = utils.convert_method_to_component(
aiplatform_sdk.Endpoint, aiplatform_sdk.Endpoint.create
)

0 comments on commit 538e4ef

Please sign in to comment.