Code

feat: Support uploading local models#779

Code

Merged

sasha-gitg merged 39 commits intogoogleapis:maingoogleapis/python-aiplatform:mainfrom Ark-kun:feat--Support-uploading-local-modelsArk-kun/python-aiplatform:feat--Support-uploading-local-modelsCopy head branch name to clipboardDec 1, 2021

+1,055

Commits

Commits on Oct 20, 2021

• feat: Support uploading local models

  Ark-kun

  committed
  bfb4f63

• feat: Added the upload_to_gcs utility function

  Show description for a62c264
  Ark-kun

  committed
  a62c264

• feat: Support staging directories in stage_local_data_in_gcs

  Show description for cf97d3b
  Ark-kun

  committed
  cf97d3b

• feat: Validate model directories before uploading

  Ark-kun

  committed
  dd35ac4

• feat: Added framework-specific model uploaders for XGBoost, Scikit-learn and Tensorflow

  Ark-kun

  committed
  d554bbf

• Added docstrings

  Ark-kun

  committed
  650b767

• Fixed the version variable in error messages

  Ark-kun

  committed
  8f61efb

Commits on Nov 5, 2021

• Only validate the model directory when using a built-in serving container

  Ark-kun

  committed
  8c5448d

• Added more exception causes to the dostrings

  Ark-kun

  committed
  4bbf2f0

• Cleaning up the temporary model directory

  Ark-kun

  committed
  171e737

• Added fallback to default credentials in upload_to_gcs

  Ark-kun

  committed
  7f5407e

Commits on Nov 13, 2021

• Moved the stage_local_data_in_gcs and upload_to_gcs functions to gcs_utils module

  Ark-kun

  committed
  33942da

• Added the "Raises" sections to function docstrings

  Ark-kun

  committed
  5febd56

• When user-provided version is not recognized, show warning message instead of error message

  Ark-kun

  committed
  439d413

• Added docstring for _get_container_registry

  Ark-kun

  committed
  1064243

• Added missing import

  Ark-kun

  committed
  03f3e80

Commits on Nov 16, 2021

• fix: Fixed code after switching from mkdtemp to TemporaryDirectory

  Show description for 31248d1
  Ark-kun

  committed
  31248d1

• Made the staging bucket name location-specific

  Show description for 9dcc39f
  Ark-kun

  committed
  9dcc39f

• Reverted the automatic temporary directory cleanup

  Show description for 24006fc
  Ark-kun

  committed
  24006fc

• Added unit tests for the Model.upload_* methods

  Ark-kun

  committed
  e8ecce7

• Added unit test for Model.upload data staging feature

  Ark-kun

  committed
  b521bde

Commits on Nov 23, 2021

• Added new container image versions

  Ark-kun

  committed
  b964224

• Moved the default model display names to the function signatures

  Ark-kun

  committed
  6039548

• Moved the default container versions to the function definitions

  Ark-kun

  committed
  079e6e9

Commits on Nov 24, 2021

• Added model upload integration test

  Ark-kun

  committed
  73a4868

Commits on Nov 29, 2021

• Merge branch 'main' into feat--Support-uploading-local-models

  sasha-gitg

  authored
  fc55618

• Merge branch 'main' into feat--Support-uploading-local-models

  sasha-gitg

  authored
  9138036

Commits on Dec 1, 2021

• Removed the container image URI generation code as asked by the reviewer

  Show description for db65f67
  Ark-kun

  committed
  db65f67

• Added the return type annotations

  Ark-kun

  committed
  4d2e440

• Removed the Optional type annotation from parameters with default value

  Ark-kun

  committed
  7b71234

• Added "vertex" to the default staging bucket name suffix

  Ark-kun

  committed
  ab87299

• Falling back to the globally-set credentials when not specified in stage_local_data_in_gcs

  Ark-kun

  committed
  6ced3ba

• Changed the static methods to class methods as requested

  Ark-kun

  committed
  7588d4b

• Auto-deleting the temporary files

  Ark-kun

  committed
  39abcc3

• Reworded some error messages

  Ark-kun

  committed
  af8317b

• Resolved the integration test feedback

  Ark-kun

  committed
  624048a

• Removed unused imports

  Ark-kun

  committed
  1ab4c9a

• Fixed the integration test teardown

  Ark-kun

  committed
  fb02072

• Merge branch 'main' into feat--Support-uploading-local-models

  sasha-gitg

  authored
  17344da