From 4adbbb3f5c4033856c00b20de6b5de715d177e38 Mon Sep 17 00:00:00 2001 From: CodemodService Bot Date: Mon, 4 Nov 2024 14:26:55 -0800 Subject: [PATCH] Update AWS IMDSv1 calls to IMDSv2 in fbcode/torchsnapshot/github/.github/workflows (#181) Summary: Pull Request resolved: https://github.com/pytorch/torchsnapshot/pull/181 Differential Revision: D65188986 --- .github/workflows/run_tests.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/run_tests.yaml b/.github/workflows/run_tests.yaml index d16e1dc..aa43eee 100644 --- a/.github/workflows/run_tests.yaml +++ b/.github/workflows/run_tests.yaml @@ -63,7 +63,7 @@ jobs: # Pulled from instance metadata endpoint for EC2 # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html category=$1 - curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" + curl -H "X-aws-ec2-metadata-token: $(curl -s -X PUT "http://169.254.169.254/latest/api/token" -H "X-aws-ec2-metadata-token-ttl-seconds: 30")" -fsSL "http://169.254.169.254/latest/meta-data/${category}" } echo "ami-id: $(get_ec2_metadata ami-id)" echo "instance-id: $(get_ec2_metadata instance-id)"