fix unit-test

Author: knikure

src/sagemaker/telemetry/telemetry_logging.py

@@ -76,7 +76,9 @@ def wrapper(*args, **kwargs):
                     "sagemaker_session", _get_default_sagemaker_session()
                 )
 
+            print("sagemaker_session: ", sagemaker_session)
             if sagemaker_session:
+                print("sagemaker_session found...")
                 logger.debug("sagemaker_session found, preparing to emit telemetry...")
                 logger.info(TELEMETRY_OPT_OUT_MESSAGING)
                 response = None
@@ -149,6 +151,7 @@ def wrapper(*args, **kwargs):
                         raise caught_ex
                     return response  # pylint: disable=W0150
             else:
+                print("sagemaker_session not found")
                 logger.debug(
                     "Unable to send telemetry for function %s. "
                     "sagemaker_session is not provided or not valid.",
@@ -185,10 +188,13 @@ def _send_telemetry_request(
             extra_info,
         )
         # Send the telemetry request
+        print(f"\nSending telemetry request to [{url}]\n")
         logger.debug("Sending telemetry request to [%s]", url)
         _requests_helper(url, 2)
+        print("Telemetry request sent...")
         logger.debug("SageMaker Python SDK telemetry successfully emitted.")
     except Exception:  # pylint: disable=W0703
+        print("SageMaker Python SDK telemetry not emitted!")
         logger.debug("SageMaker Python SDK telemetry not emitted!")
 
 

tests/unit/sagemaker/telemetry/test_telemetry_logging.py

@@ -289,7 +289,7 @@ def test_get_region_or_default_exception(self):
     def test_get_default_sagemaker_session(self):
         sagemaker_session = _get_default_sagemaker_session()
 
-        assert sagemaker_session is sagemaker.Session
+        assert isinstance(sagemaker_session, sagemaker.Session) == True
         assert sagemaker_session.boto_session.region_name == "us-west-2"
 
     @patch.object(boto3.Session, "region_name", None)