chore: log info msg when overriding jumpstart bucket

Author: evakravi

src/sagemaker/jumpstart/utils.py

@@ -66,7 +66,9 @@ def get_jumpstart_content_bucket(region: str) -> str:
         constants.ENV_VARIABLE_JUMPSTART_CONTENT_BUCKET_OVERRIDE in os.environ
         and len(os.environ[constants.ENV_VARIABLE_JUMPSTART_CONTENT_BUCKET_OVERRIDE]) > 0
     ):
-        return os.environ[constants.ENV_VARIABLE_JUMPSTART_CONTENT_BUCKET_OVERRIDE]
+        bucket_override = os.environ[constants.ENV_VARIABLE_JUMPSTART_CONTENT_BUCKET_OVERRIDE]
+        LOGGER.info("Using JumpStart bucket override: '%s'", bucket_override)
+        return bucket_override
     try:
         return constants.JUMPSTART_REGION_NAME_TO_LAUNCHED_REGION_DICT[region].content_bucket
     except KeyError:

tests/unit/sagemaker/jumpstart/test_utils.py

@@ -44,8 +44,13 @@ def test_get_jumpstart_content_bucket():
 
 def test_get_jumpstart_content_bucket_override():
     with patch.dict(os.environ, {ENV_VARIABLE_JUMPSTART_CONTENT_BUCKET_OVERRIDE: "some-val"}):
-        random_region = "random_region"
-        assert "some-val" == utils.get_jumpstart_content_bucket(random_region)
+        with patch("logging.Logger.info") as mocked_info_log:
+            random_region = "random_region"
+            assert "some-val" == utils.get_jumpstart_content_bucket(random_region)
+            mocked_info_log.assert_called_once_with(
+                "Using JumpStart bucket override: '%s'",
+                "some-val",
+            )
 
 
 def test_get_jumpstart_launched_regions_message():