AWS SageMaker(禁用互联网)无法连接到 STS

AWS SageMaker (with internet disabled) unable to connect to STS

我无法执行简单的操作:

import sagemaker
sess = sagemaker.Session()
role = sagemaker.get_execution_role()

因为我的笔记本实例没有连接到互联网。我在与笔记本实例相同的子网中有一个 STS 端点接口,但我认为 sagemaker API 正在使用全局端点。一段时间后我实际上收到以下错误消息:

ConnectTimeoutError: Connect timeout on endpoint URL: "https://sts.us-east-1.amazonaws.com/"

我该如何解决这个问题?或者是否需要更新 sagemaker 模块?

所以解决方案是为 sagemaker API(api.sagemaker...)和 STS 添加一个 VPC 端点。