Skip to content

Commit 534270e

Browse files
committed
Rename destroy method
1 parent 2f13f3d commit 534270e

File tree

6 files changed

+6
-10
lines changed

6 files changed

+6
-10
lines changed

src/sagemaker/model.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def deploy(self, *args, **kwargs) -> PredictorBase:
5252
"""Deploy this model to a compute service."""
5353

5454
@abc.abstractmethod
55-
def destroy(self, *args, **kwargs) -> None:
55+
def delete_model(self, *args, **kwargs) -> None:
5656
"""Destroy resources associated with this model."""
5757

5858

@@ -823,8 +823,6 @@ def delete_model(self):
823823
)
824824
self.sagemaker_session.delete_model(self.name)
825825

826-
destroy = delete_model
827-
828826

829827
SCRIPT_PARAM_NAME = "sagemaker_program"
830828
DIR_PARAM_NAME = "sagemaker_submit_directory"

src/sagemaker/predictor.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def predict(self, *args, **kwargs) -> Any:
6262
"""Perform inference on the provided data and return a prediction."""
6363

6464
@abc.abstractmethod
65-
def destroy(self, *args, **kwargs) -> None:
65+
def delete_endpoint(self, *args, **kwargs) -> None:
6666
"""Destroy resources associated with this predictor."""
6767

6868
@property
@@ -330,8 +330,6 @@ def delete_endpoint(self, delete_endpoint_config=True):
330330

331331
self.sagemaker_session.delete_endpoint(self.endpoint_name)
332332

333-
destroy = delete_endpoint
334-
335333
def delete_model(self):
336334
"""Deletes the Amazon SageMaker models backing this predictor."""
337335
request_failed = False

src/sagemaker/serverless/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def deploy(
8282

8383
return LambdaPredictor(function_name, client=self._client)
8484

85-
def destroy(self) -> None:
85+
def delete_model(self) -> None:
8686
"""Destroy resources associated with this model.
8787
8888
This method does not delete the image specified in the constructor. As

src/sagemaker/serverless/predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def predict(self, data: dict) -> dict:
6161
response["ResponseMetadata"]["HTTPHeaders"]["content-type"],
6262
)
6363

64-
def destroy(self) -> None:
64+
def delete_endpoint(self) -> None:
6565
"""Destroy the Lambda function specified in the constructor."""
6666
self._client.delete_function(FunctionName=self._function_name)
6767

tests/unit/sagemaker/serverless/test_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,4 +49,4 @@ def test_deploy(mock_client, wait):
4949

5050
def test_destroy():
5151
model = LambdaModel(IMAGE_URI, ROLE, client=mock_client)
52-
model.destroy() # NOTE: This method is a no-op.
52+
model.delete_model() # NOTE: This method is a no-op.

tests/unit/sagemaker/serverless/test_predictor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_predict(mock_client):
4848
def test_destroy(mock_client):
4949
predictor = LambdaPredictor(FUNCTION_NAME, client=mock_client)
5050

51-
predictor.destroy()
51+
predictor.delete_endpoint()
5252

5353
mock_client.delete_function.assert_called_once()
5454
_, kwargs = mock_client.delete_function.call_args

0 commit comments

Comments
 (0)