fix(local-image-aws-creds): use short lived creds

Author: wcarpenter1-godaddy

src/sagemaker/local/image.py

@@ -1014,7 +1014,7 @@ def _aws_credentials(session):
                 "AWS_ACCESS_KEY_ID=%s" % (str(access_key)),
                 "AWS_SECRET_ACCESS_KEY=%s" % (str(secret_key)),
             ]
-        if not _aws_credentials_available_in_metadata_service():
+        if _use_short_lived_credentials() or not _aws_credentials_available_in_metadata_service():
             logger.warning(
                 "Using the short-lived AWS credentials found in session. They might expire while "
                 "running."
@@ -1052,6 +1052,10 @@ def _aws_credentials_available_in_metadata_service():
     return not instance_metadata_provider.load() is None
 
 
+def _use_short_lived_credentials():
+    return os.environ.get("USE_SHORT_LIVED_CREDENTIALS") == "1"
+
+
 def _write_json_file(filename, content):
     """Write the contents dict as json to the file.