@ericpark3 the type-hint is incorrect in this line. https://github.com/aws/sagemaker-python-sdk/blame/1ae7ce938e9b94aeb421b7af1570c6a7c3ae4a6e/src/sagemaker/feature_store/feature_group.py#L280