Complete the airflow survey & get a free airflow 3 certification!

tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service

Example Airflow DAG for Google Vertex AI service testing Model Service operations.

Attributes

ENV_ID

PROJECT_ID

DAG_ID

REGION

TRAIN_DISPLAY_NAME

MODEL_DISPLAY_NAME

RESOURCE_DATA_BUCKET

DATA_SAMPLE_GCS_BUCKET_NAME

STAGING_BUCKET

DATA_SAMPLE_GCS_OBJECT_NAME

TABULAR_DATASET

CONTAINER_URI

IS_COMPOSER

LOCAL_TRAINING_SCRIPT_PATH

MODEL_OUTPUT_CONFIG

MODEL_SERVING_CONTAINER_URI

MODEL_OBJ

MODEL_OBJ_V2

create_bucket

test_run

Module Contents

tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.ENV_ID[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.PROJECT_ID[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.DAG_ID = 'vertex_ai_model_service_operations'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.REGION = 'us-central1'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.TRAIN_DISPLAY_NAME = 'train-housing-custom-Uninferable'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_DISPLAY_NAME = 'custom-housing-model-Uninferable'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.RESOURCE_DATA_BUCKET = 'airflow-system-tests-resources'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.DATA_SAMPLE_GCS_BUCKET_NAME = ''[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.STAGING_BUCKET = 'gs://'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.DATA_SAMPLE_GCS_OBJECT_NAME = 'vertex-ai/california_housing_train.csv'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.TABULAR_DATASET[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.CONTAINER_URI = 'us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.IS_COMPOSER[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.LOCAL_TRAINING_SCRIPT_PATH = 'gcs/data/california_housing_training_script.py'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_OUTPUT_CONFIG[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_SERVING_CONTAINER_URI = 'us-docker.pkg.dev/vertex-ai/prediction/tf2-cpu.2-2:latest'[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_OBJ[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_OBJ_V2[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.create_bucket[source]
tests.system.google.cloud.vertex_ai.example_vertex_ai_model_service.test_run[source]

Was this entry helpful?