diff --git a/1.architectures/5.sagemaker-hyperpod/easy-ssh.sh b/1.architectures/5.sagemaker-hyperpod/easy-ssh.sh index 29770c82..0e2a1d3c 100755 --- a/1.architectures/5.sagemaker-hyperpod/easy-ssh.sh +++ b/1.architectures/5.sagemaker-hyperpod/easy-ssh.sh @@ -142,7 +142,7 @@ add_keypair_to_cluster echo -e "\nNow you can run:\n" echo -e "$ ${GREEN}ssh ${cluster_name}${NC}" -[[ DRY_RUN -eq 1 ]] && exit 0 +[[ DRY_RUN -eq 1 ]] && echo -e "\n${GREEN}aws ssm start-session "${aws_cli_args[@]}" --target sagemaker-cluster:${cluster_id}_${node_group}-${instance_id}${NC}\n" && exit 0 # Start session as Ubuntu only if the SSM-SessionManagerRunShellAsUbuntu document exists. if aws ssm describe-document "${aws_cli_args[@]}" --name SSM-SessionManagerRunShellAsUbuntu > /dev/null 2>&1; then