Skip to content

Commit fab0196

Browse files
committed
Rename delete_endpoint as delete_predictor
1 parent adcbbaf commit fab0196

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

src/sagemaker/predictor.py

+3-1
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 delete_endpoint(self, *args, **kwargs) -> None:
65+
def delete_predictor(self, *args, **kwargs) -> None:
6666
"""Destroy resources associated with this predictor."""
6767

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

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

333+
delete_predictor = delete_endpoint
334+
333335
def delete_model(self):
334336
"""Deletes the Amazon SageMaker models backing this predictor."""
335337
request_failed = False

src/sagemaker/serverless/predictor.py

+1-1
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 delete_endpoint(self) -> None:
64+
def delete_predictor(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_predictor.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_predict(mock_client):
4848
def test_delete_endpoint(mock_client):
4949
predictor = LambdaPredictor(FUNCTION_NAME, client=mock_client)
5050

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

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

0 commit comments

Comments
 (0)