class SagemakerClient:
def __init__(self):
self.client = Session(profile_name="default").client(
"sagemaker", region_name="us-west-2")
def create_model(self, model_data_url):
model_params = {
"ExecutionRoleArn": "arn:aws:iam::562998738767:role/dev-sagemaker",
"ModelName": "sample-model5",
"PrimaryContainer": {
"Image":
"562998738767.dkr.ecr.us-west-2.amazonaws.com/test-sagemaker:latest",
"ModelDataUrl": model_data_url
}
}
self.client.create_model(**model_params)
