airflow.providers.google.cloud.operators.mlengine

This module contains Google Cloud MLEngine operators.

Attributes

log

Classes

MLEngineCreateModelOperator

Creates a new model.

Module Contents

airflow.providers.google.cloud.operators.mlengine.log[source]
class airflow.providers.google.cloud.operators.mlengine.MLEngineCreateModelOperator(*, model, project_id=PROVIDE_PROJECT_ID, gcp_conn_id='google_cloud_default', impersonation_chain=None, **kwargs)[source]

Bases: airflow.providers.google.cloud.operators.cloud_base.GoogleCloudBaseOperator

Creates a new model.

Warning

This operator is deprecated. Please use appropriate VertexAI operator from airflow.providers.google.cloud.operators.vertex_ai instead.

See also

For more information on how to use this operator, take a look at the guide: Creating a model

The model should be provided by the model parameter.

Parameters:

  • model (dict) – A dictionary containing the information about the model.

  • project_id (str) – The Google Cloud project name to which MLEngine model belongs. If set to None or missing, the default project_id from the Google Cloud connection is used. (templated)

  • gcp_conn_id (str) – The connection ID to use when fetching connection info.

  • impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials, or chained list of accounts required to get the access_token of the last account in the list, which will be impersonated in the request. If set as a string, the account must grant the originating account the Service Account Token Creator IAM role. If set as a sequence, the identities from the list must grant Service Account Token Creator IAM role to the directly preceding identity, with first account from the list granting this role to the originating account (templated).

template_fields: collections.abc.Sequence[str] = ('project_id', 'model', 'impersonation_chain')[source]
project_id = None[source]
model[source]
impersonation_chain = None[source]
execute(context)[source]

Derive when creating an operator.

Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

Was this entry helpful?