From 10026e3249cf87a51047b29ab55d8ebb27bae9a8 Mon Sep 17 00:00:00 2001 From: dberenbaum Date: Mon, 9 Oct 2023 15:35:45 -0400 Subject: [PATCH] deletes model endpoint to avoid errors deploying to existing endpoint --- .../code/sagemaker/deploy_model.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/example-get-started-experiments/code/sagemaker/deploy_model.py b/example-get-started-experiments/code/sagemaker/deploy_model.py index da751d17..ea5c651b 100644 --- a/example-get-started-experiments/code/sagemaker/deploy_model.py +++ b/example-get-started-experiments/code/sagemaker/deploy_model.py @@ -2,6 +2,8 @@ import re import sys +import boto3 + from sagemaker.deserializers import JSONDeserializer from sagemaker.pytorch import PyTorchModel from sagemaker.serverless import ServerlessInferenceConfig @@ -32,9 +34,14 @@ def deploy( sagemaker_logger.setLevel(logging.DEBUG) sagemaker_logger.addHandler(logging.StreamHandler(sys.stdout)) - composed_name = re.sub( + composed_name = re.sub( r"[^a-zA-Z0-9\-]", "-", f"{name}-{version}-{stage}") + try: + boto3.client("sagemaker").delete_endpoint(EndpointName=composed_name) + except: + pass + model = PyTorchModel( name=composed_name, model_data=model_data,