fix: update sagemaker.serverless integration test

src/sagemaker/predictor.py

@@ -62,7 +62,7 @@ def predict(self, *args, **kwargs) -> Any:
         """Perform inference on the provided data and return a prediction."""
 
     @abc.abstractmethod
-    def delete_endpoint(self, *args, **kwargs) -> None:
+    def delete_predictor(self, *args, **kwargs) -> None:
         """Destroy resources associated with this predictor."""
 
     @property
@@ -330,6 +330,8 @@ def delete_endpoint(self, delete_endpoint_config=True):
 
         self.sagemaker_session.delete_endpoint(self.endpoint_name)
 
+    delete_predictor = delete_endpoint
+
     def delete_model(self):
         """Deletes the Amazon SageMaker models backing this predictor."""
         request_failed = False

src/sagemaker/serverless/predictor.py

@@ -61,7 +61,7 @@ def predict(self, data: dict) -> dict:
             response["ResponseMetadata"]["HTTPHeaders"]["content-type"],
         )
 
-    def delete_endpoint(self) -> None:
+    def delete_predictor(self) -> None:
         """Destroy the Lambda function specified in the constructor."""
         self._client.delete_function(FunctionName=self._function_name)
 

tests/integ/test_serverless.py

@@ -12,25 +12,25 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-import os
-
+import boto3
 import pytest
 
 from sagemaker.serverless import LambdaModel
 from sagemaker.utils import unique_name_from_base
 
 # See tests/data/serverless for the image source code.
-IMAGE_URI = "142577830533.dkr.ecr.us-west-2.amazonaws.com/serverless-integ-test:latest"
-ROLE = "arn:aws:iam::142577830533:role/lambda_basic_execution"
-URL = "https://c.files.bbci.co.uk/12A9B/production/_111434467_gettyimages-1143489763.jpg"
+ACCOUNT_ID = 142577830533
+IMAGE_URI = f"{ACCOUNT_ID}.dkr.ecr.us-west-2.amazonaws.com/serverless-integ-test:latest"
+ROLE = f"arn:aws:iam::{ACCOUNT_ID}:role/lambda_basic_execution"
+URL = "https://sagemaker-integ-tests-data.s3.us-east-1.amazonaws.com/cat.jpeg"
 
 
-@pytest.mark.skipif(
-    "CODEBUILD_BUILD_ID" not in os.environ,
-    reason="The container image is private to the CI account.",
-)
 def test_lambda():
-    model = LambdaModel(image_uri=IMAGE_URI, role=ROLE)
+    client = boto3.client("lambda")
+    if client.get_caller_identity().get("Account") != ACCOUNT_ID:
+        pytest.skip("The container image is private to the CI account.")
+
+    model = LambdaModel(image_uri=IMAGE_URI, role=ROLE, client=client)
 
     predictor = model.deploy(
         unique_name_from_base("my-lambda-function"), timeout=60, memory_size=4092
@@ -39,5 +39,5 @@ def test_lambda():
 
     assert prediction == {"class": "tabby"}
 
-    model.destroy()
-    predictor.destroy()
+    model.delete_model()
+    predictor.delete_predictor()

tests/unit/sagemaker/serverless/test_predictor.py

@@ -48,7 +48,7 @@ def test_predict(mock_client):
 def test_delete_endpoint(mock_client):
     predictor = LambdaPredictor(FUNCTION_NAME, client=mock_client)
 
-    predictor.delete_endpoint()
+    predictor.delete_predictor()
 
     mock_client.delete_function.assert_called_once()
     _, kwargs = mock_client.delete_function.call_args