tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job¶
Example Airflow DAG for Google Vertex AI service testing Custom Jobs operations.
Module Contents¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.DAG_ID = 'vertex_ai_custom_job_operations'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.REGION = 'us-central1'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.CUSTOM_DISPLAY_NAME[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.MODEL_DISPLAY_NAME[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.CUSTOM_GCS_BUCKET_NAME[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.DATA_SAMPLE_GCS_OBJECT_NAME = 'vertex-ai/california_housing_train.csv'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.CSV_ZIP_FILE_LOCAL_PATH[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.CSV_FILE_LOCAL_PATH = '/custom-job/california_housing_train.csv'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.CONTAINER_URI = 'gcr.io/cloud-aiplatform/training/tf-cpu.2-2:latest'[source]¶
- tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_job.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_custom_job.REPLICA_COUNT = 1[source]¶