tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job¶

Example Airflow DAG for Google Vertex AI service testing Custom Jobs operations.

Attributes¶

`ENV_ID`
`PROJECT_ID`
`REGION`
`CUSTOM_DISPLAY_NAME`
`MODEL_DISPLAY_NAME`
`DAG_ID`
`RESOURCE_DATA_BUCKET`
`CUSTOM_GCS_BUCKET_NAME`
`DATA_SAMPLE_GCS_OBJECT_NAME`
`CONTAINER_URI`
`MODEL_SERVING_CONTAINER_URI`
`REPLICA_COUNT`
`IS_COMPOSER`
`LOCAL_TRAINING_SCRIPT_PATH`
`create_bucket`
`test_run`

Functions¶

TABULAR_DATASET(bucket_name)

Module Contents¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.ENV_ID[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.PROJECT_ID[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.REGION = 'us-central1'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.CUSTOM_DISPLAY_NAME[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.MODEL_DISPLAY_NAME[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.DAG_ID = 'vertex_ai_custom_job_operations'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.RESOURCE_DATA_BUCKET = 'airflow-system-tests-resources'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.CUSTOM_GCS_BUCKET_NAME[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.DATA_SAMPLE_GCS_OBJECT_NAME = 'vertex-ai/california_housing_train.csv'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.TABULAR_DATASET(bucket_name)[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.CONTAINER_URI = 'us-docker.pkg.dev/vertex-ai/training/tf-cpu.2-2:latest'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.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_custom_job.REPLICA_COUNT = 1[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.IS_COMPOSER[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.LOCAL_TRAINING_SCRIPT_PATH = 'gcs/data/california_housing_training_script.py'[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.create_bucket[source]¶

tests.system.google.cloud.vertex_ai.example_vertex_ai_custom_job.test_run[source]¶

Was this entry helpful?