Fix import error in integ tests and update PR check#134

zhaoqizqwang merged 2 commits intoaws:mainaws/sagemaker-hyperpod-cli:mainfrom

zhaoqizqwang:mainzhaoqizqwang/sagemaker-hyperpod-cli:mainCopy head branch name to clipboard