`tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container`¶

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

Module Contents¶

`ENV_ID`
`PROJECT_ID`
`DAG_ID`
`REGION`
`CONTAINER_DISPLAY_NAME`
`MODEL_DISPLAY_NAME`
`RESOURCE_DATA_BUCKET`
`CUSTOM_CONTAINER_GCS_BUCKET_NAME`
`DATA_SAMPLE_GCS_OBJECT_NAME`
`CONTAINER_URI`
`CUSTOM_CONTAINER_URI`
`MODEL_SERVING_CONTAINER_URI`
`REPLICA_COUNT`
`MACHINE_TYPE`
`ACCELERATOR_TYPE`
`ACCELERATOR_COUNT`
`TRAINING_FRACTION_SPLIT`
`TEST_FRACTION_SPLIT`
`VALIDATION_FRACTION_SPLIT`
`create_bucket`
`test_run`

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.ENV_ID[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.PROJECT_ID[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.DAG_ID = 'example_vertex_ai_custom_job_operations'[source]¶

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

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.CONTAINER_DISPLAY_NAME[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.MODEL_DISPLAY_NAME[source]¶

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

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.CUSTOM_CONTAINER_GCS_BUCKET_NAME[source]¶

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

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

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.CONTAINER_URI = 'gcr.io/cloud-aiplatform/training/tf-cpu.2-2:latest'[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.CUSTOM_CONTAINER_URI[source]¶

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

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.MACHINE_TYPE = 'n1-standard-4'[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.ACCELERATOR_TYPE = 'ACCELERATOR_TYPE_UNSPECIFIED'[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.ACCELERATOR_COUNT = 0[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.TRAINING_FRACTION_SPLIT = 0.7[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.TEST_FRACTION_SPLIT = 0.15[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.VALIDATION_FRACTION_SPLIT = 0.15[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.create_bucket[source]¶

tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.test_run[source]¶