pipelines/components/aws/sagemaker/tests/integration_tests/utils/sagemaker_utils.py

27 lines
798 B
Python

def describe_training_job(client, training_job_name):
return client.describe_training_job(TrainingJobName=training_job_name)
def describe_model(client, model_name):
return client.describe_model(ModelName=model_name)
def describe_endpoint(client, endpoint_name):
return client.describe_endpoint(EndpointName=endpoint_name)
def delete_endpoint(client, endpoint_name):
client.delete_endpoint(EndpointName=endpoint_name)
waiter = client.get_waiter("endpoint_deleted")
waiter.wait(EndpointName=endpoint_name)
def describe_hpo_job(client, job_name):
return client.describe_hyper_parameter_tuning_job(
HyperParameterTuningJobName=job_name
)
def describe_transform_job(client, job_name):
return client.describe_transform_job(TransformJobName=job_name)