Skip to content

Commit dadbb22

Browse files
author
Roja Reddy Sareddy
committed
change: Allow telemetry only in supported regions
1 parent f972222 commit dadbb22

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

src/sagemaker/telemetry/telemetry_logging.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -190,19 +190,16 @@ def _send_telemetry_request(
190190
"""Make GET request to an empty object in S3 bucket"""
191191
try:
192192
accountId = _get_accountId(session) if session else "NotAvailable"
193+
region = _get_region_or_default(session)
194+
195+
try:
196+
Region(region) # Validate the region
197+
except ValueError:
198+
logger.warning(
199+
"Region not found in supported regions. Telemetry request will not be emitted."
200+
)
201+
return
193202

194-
# Validate region if session exists
195-
if session:
196-
region = _get_region_or_default(session)
197-
try:
198-
Region(region)
199-
except ValueError:
200-
logger.warning(
201-
"Region not found in supported regions. Telemetry request will not be emitted."
202-
)
203-
return
204-
else: # telemetry will be sent to us-west-2 if no session available
205-
region = DEFAULT_AWS_REGION
206203
url = _construct_url(
207204
accountId,
208205
region,
@@ -280,6 +277,7 @@ def _get_region_or_default(session):
280277

281278
def _get_default_sagemaker_session():
282279
"""Return the default sagemaker session"""
280+
283281
boto_session = boto3.Session(region_name=DEFAULT_AWS_REGION)
284282
sagemaker_session = Session(boto_session=boto_session)
285283

0 commit comments

Comments
 (0)