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

tfx.extensions.google_cloud_ai_platform.runner.create_model_for_aip_prediction_if_not_exist

Creates a new model for serving with AI Platform if not exists.

tfx.extensions.google_cloud_ai_platform.runner.create_model_for_aip_prediction_if_not_exist(
    api: discovery.Resource,
    job_labels: Dict[Text, Text],
    ai_platform_serving_args: Dict[Text, Any]
) -> bool

api Google API client resource.
job_labels The dict of labels that will be attached to this job.
ai_platform_serving_args Dictionary containing arguments for pushing to AI Platform.

Whether a new model is created.

RuntimeError if model creation failed.