"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),
),
