Join the SIG TFX-Addons community and help make TFX even better!


Deploys a model for serving with AI Platform.

api Google API client resource.
serving_path The path to the model. Must be a GCS URI.
model_version Version of the model being deployed. Must be different from what is currently being served.
ai_platform_serving_args Dictionary containing arguments for pushing to AI Platform. The full set of parameters supported can be found at Most keys are forwarded as-is, but following keys are handled specially:

  • name: this must be empty (and will be filled by pusher).
  • deployment_uri: this must be empty (and will be filled by pusher).
  • python_version: when left empty, this will be filled by python version of the environment being used.
  • runtime_version: when left empty, this will be filled by TensorFlow version from the environment.
  • labels: a list of job labels will be merged with user's input.
job_labels The dict of labels that will be attached to this job. They are merged with optional labels from ai_platform_serving_args.
skip_model_creation If true, the method assuem model already exist in AI platform, therefore skipping model creation.
set_default_version Whether set the newly deployed model version as the default version.

RuntimeError if an error is encountered when trying to push.