feat: Add Custom Container Prediction support, move to single API endpoint #277

vinnysenthil · 2021-03-31T05:34:44Z

Summary of Changes

Make artifact_uri optional in Model.upload() to allow for custom containers that contain model artifacts & server
Remove reference to specific prediction endpoint (*-prediction-aiplatform.googleapis.com) - link to commit
Update relevant tests

sasha-gitg

LGTM! Thanks!

google/cloud/aiplatform/models.py

vinnysenthil added 3 commits March 30, 2021 18:26

Make artifact_uri optional in Model.upload

827216d

Update Model.upload test for optional artifact_uri

5887c9d

Move prediction to core API endpoint (b/180036930)

bbd9218

vinnysenthil requested review from ivanmkc, morgandu and sasha-gitg March 31, 2021 05:34

google-cla bot added the cla: yes This human has signed the Contributor License Agreement. label Mar 31, 2021

sasha-gitg approved these changes Mar 31, 2021

View reviewed changes

google/cloud/aiplatform/models.py Outdated Show resolved Hide resolved

Update typehint to optional

1106356

vinnysenthil added the automerge Merge the pull request once unit tests and other checks pass. label Mar 31, 2021

gcf-merge-on-green bot merged commit ca7f6d6 into dev Mar 31, 2021

gcf-merge-on-green bot deleted the mb-custom-prediction branch March 31, 2021 15:36

gcf-merge-on-green bot removed the automerge Merge the pull request once unit tests and other checks pass. label Mar 31, 2021