tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service

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

Module Contents

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.ENV_ID[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.PROJECT_ID[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.DAG_ID = 'example_vertex_ai_model_service_operations'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.REGION = 'us-central1'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.TRAIN_DISPLAY_NAME[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_DISPLAY_NAME[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.RESOURCE_DATA_BUCKET = 'airflow-system-tests-resources'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.DATA_SAMPLE_GCS_BUCKET_NAME[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.STAGING_BUCKET[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.DATA_SAMPLE_GCS_OBJECT_NAME = 'vertex-ai/california_housing_train.csv'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.TABULAR_DATASET[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.CONTAINER_URI = 'gcr.io/cloud-aiplatform/training/tf-cpu.2-2:latest'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.LOCAL_TRAINING_SCRIPT_PATH = 'california_housing_training_script.py'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_OUTPUT_CONFIG[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_SERVING_CONTAINER_URI = 'gcr.io/cloud-aiplatform/prediction/tf2-cpu.2-2:latest'[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.MODEL_OBJ[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.create_bucket[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_model_service.test_run[source]

Was this entry helpful?