"boot_disk_type": "pd-ssd", "boot_disk_size_gb": 100 }, "container_spec": { "image_uri": _TEST_TRAINING_CONTAINER_IMAGE, "command": [], "args": _TEST_RUN_ARGS, }, }] _TEST_STAGING_BUCKET = "gs://test-staging-bucket" _TEST_BASE_OUTPUT_DIR = f"{_TEST_STAGING_BUCKET}/{_TEST_DISPLAY_NAME}" # CMEK encryption _TEST_DEFAULT_ENCRYPTION_KEY_NAME = "key_default" _TEST_DEFAULT_ENCRYPTION_SPEC = gca_encryption_spec_compat.EncryptionSpec( kms_key_name=_TEST_DEFAULT_ENCRYPTION_KEY_NAME) _TEST_SERVICE_ACCOUNT = "*****@*****.**" _TEST_NETWORK = f"projects/{_TEST_PROJECT}/global/networks/{_TEST_ID}" _TEST_TIMEOUT = 8000 _TEST_RESTART_JOB_ON_WORKER_RESTART = True _TEST_LABELS = {"my_key": "my_value"} _TEST_BASE_CUSTOM_JOB_PROTO = gca_custom_job_compat.CustomJob( display_name=_TEST_DISPLAY_NAME, job_spec=gca_custom_job_compat.CustomJobSpec( worker_pool_specs=_TEST_WORKER_POOL_SPEC, base_output_directory=gca_io_compat.GcsDestination(
}) }, outputs={ "medv": explain.ExplanationMetadata.OutputMetadata( {"output_tensor_name": "dense_2"}) }, ) _TEST_EXPLANATION_PARAMETERS = explain.ExplanationParameters( {"sampled_shapley_attribution": { "path_count": 10 }}) # CMEK encryption _TEST_ENCRYPTION_KEY_NAME = "key_1234" _TEST_ENCRYPTION_SPEC = gca_encryption_spec.EncryptionSpec( kms_key_name=_TEST_ENCRYPTION_KEY_NAME) _TEST_ENDPOINT_GAPIC = gca_endpoint.Endpoint(display_name=_TEST_DISPLAY_NAME, name=_TEST_ENDPOINT_NAME) _TEST_ENDPOINT_LIST = [ gca_endpoint.Endpoint( name=_TEST_ENDPOINT_NAME, display_name="aac", create_time=datetime.now() - timedelta(minutes=15), ), gca_endpoint.Endpoint( name=_TEST_ENDPOINT_NAME, display_name="aab", create_time=datetime.now() - timedelta(minutes=5), ),