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

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 = 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.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.CSV_FILE_LOCAL_PATH = /custom-job/california_housing_train.csv[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.RESOURCES_PATH[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.CSV_ZIP_FILE_LOCAL_PATH[source]
tests.system.providers.google.cloud.vertex_ai.example_vertex_ai_custom_container.TABULAR_DATASET[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]

Was this entry helpful?