airflow.providers.google.cloud.sensors.vertex_ai.feature_store¶

This module contains a Vertex AI Feature Store sensor.

Classes¶

FeatureViewSyncSensor

Sensor to monitor the state of a Vertex AI Feature View sync operation.

Module Contents¶

class airflow.providers.google.cloud.sensors.vertex_ai.feature_store.FeatureViewSyncSensor(*, feature_view_sync_name, location, gcp_conn_id='google_cloud_default', wait_timeout=None, impersonation_chain=None, **kwargs)[source]¶

Bases: airflow.providers.common.compat.sdk.BaseSensorOperator

Sensor to monitor the state of a Vertex AI Feature View sync operation.

Parameters:

feature_view_sync_name (str) – The name of the feature view sync operation to monitor. (templated)
location (str) – Required. The Cloud region in which to handle the request. (templated)
gcp_conn_id (str) – The connection ID to use connecting to Google Cloud Platform.
wait_timeout (int | None) – How many seconds to wait for sync to complete.
impersonation_chain (str | collections.abc.Sequence[str] | None) – Optional service account to impersonate using short-term credentials.

template_fields: collections.abc.Sequence[str] = ('location', 'feature_view_sync_name')[source]¶

ui_color = '#f0eee4'[source]¶

feature_view_sync_name[source]¶

location[source]¶

gcp_conn_id = 'google_cloud_default'[source]¶

wait_timeout = None[source]¶

impersonation_chain = None[source]¶

start_sensor_time: float | None = None[source]¶

execute(context)[source]¶

Derive when creating an operator.

The main method to execute the task. Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

poke(context)[source]¶

Override when deriving this class.