From 876dd80d17a85ecbfeea21359876fc5a426ca4d0 Mon Sep 17 00:00:00 2001 From: Evan Kravitz Date: Thu, 7 Sep 2023 12:03:43 +0000 Subject: [PATCH 1/2] fix: handle bad jumpstart default session --- src/sagemaker/jumpstart/constants.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/sagemaker/jumpstart/constants.py b/src/sagemaker/jumpstart/constants.py index ad764317c6..171d1026be 100644 --- a/src/sagemaker/jumpstart/constants.py +++ b/src/sagemaker/jumpstart/constants.py @@ -179,6 +179,13 @@ JUMPSTART_LOGGER = logging.getLogger("sagemaker.jumpstart") -DEFAULT_JUMPSTART_SAGEMAKER_SESSION = Session( - boto3.Session(region_name=JUMPSTART_DEFAULT_REGION_NAME) -) +try: + DEFAULT_JUMPSTART_SAGEMAKER_SESSION = Session( + boto3.Session(region_name=JUMPSTART_DEFAULT_REGION_NAME) + ) +except Exception as e: + DEFAULT_JUMPSTART_SAGEMAKER_SESSION = None + JUMPSTART_LOGGER.warn( + "Unable to create default JumpStart SageMaker Session due to the following error: %s.", + str(e), + ) From ad473c81caa91d4550da9d6c13f3e3188db23305 Mon Sep 17 00:00:00 2001 From: Evan Kravitz Date: Thu, 7 Sep 2023 12:21:15 +0000 Subject: [PATCH 2/2] fix: pylint issues --- src/sagemaker/jumpstart/constants.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/sagemaker/jumpstart/constants.py b/src/sagemaker/jumpstart/constants.py index 171d1026be..bf6a6c264a 100644 --- a/src/sagemaker/jumpstart/constants.py +++ b/src/sagemaker/jumpstart/constants.py @@ -183,9 +183,9 @@ DEFAULT_JUMPSTART_SAGEMAKER_SESSION = Session( boto3.Session(region_name=JUMPSTART_DEFAULT_REGION_NAME) ) -except Exception as e: +except Exception as e: # pylint: disable=W0703 DEFAULT_JUMPSTART_SAGEMAKER_SESSION = None - JUMPSTART_LOGGER.warn( + JUMPSTART_LOGGER.warning( "Unable to create default JumpStart SageMaker Session due to the following error: %s.", str(e), )