Update test_serverless.py

Author: bveeramani

tests/integ/test_serverless.py

@@ -20,13 +20,13 @@
 URL = "https://sagemaker-integ-tests-data.s3.us-east-1.amazonaws.com/cat.jpeg"
 
 REPOSITORY_NAME = "serverless-integ-test"
+REPOSITORY_REGION = "us-west-2"
 ROLE_NAME = "LambdaExecutionRole"
-REGION = "us-west-2"
 
 
 @pytest.fixture(name="image_uri", scope="module")
 def fixture_image_uri(account):
-    return f"{account}.dkr.ecr.{REGION}.amazonaws.com/{REPOSITORY_NAME}:latest"
+    return f"{account}.dkr.ecr.{REPOSITORY_REGION}.amazonaws.com/{REPOSITORY_NAME}:latest"
 
 
 @pytest.fixture(name="role", scope="module")
@@ -41,9 +41,12 @@ def fixture_client(boto_session):
 
 @pytest.fixture(name="repository_exists", scope="module")
 def fixture_repository_exists(boto_session):
-    client = boto_session.client("ecr")
-    client.describe_repositories(repositoryNames=[REPOSITORY_NAME])
-    return True
+    client = boto_session.client("ecr", region_name=REPOSITORY_REGION)
+    try:
+        client.describe_repositories(repositoryNames=[REPOSITORY_NAME])
+        return True
+    except client.exceptions.RepositoryNotFoundException:
+        return False
 
 
 def test_lambda(image_uri, role, client, repository_exists):