Skip to content

Commit

Permalink
no region
Browse files Browse the repository at this point in the history
  • Loading branch information
arianvp committed Mar 2, 2024
1 parent d6b7e4e commit 9a4ab6f
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions upload-ami/src/upload_ami/smoke_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,8 @@
import logging


def smoke_test(image_id, region, run_id, cancel):
ec2 = boto3.client("ec2", region_name=region)
def smoke_test(image_id, run_id, cancel):
ec2 = boto3.client("ec2")

images = ec2.describe_images(Owners=["self"], ImageIds=[image_id])
assert len(images["Images"]) == 1
Expand Down Expand Up @@ -63,12 +63,11 @@ def main():

parser = argparse.ArgumentParser()
parser.add_argument("--image-id", required=True)
parser.add_argument("--region", required=True)
parser.add_argument("--run-id", required=False)
parser.add_argument("--cancel", action="store_true", required=False)
args = parser.parse_args()

smoke_test(args.image_id, args.region, args.run_id, args.cancel)
smoke_test(args.image_id, args.run_id, args.cancel)


if __name__ == "__main__":
Expand Down

0 comments on commit 9a4ab6f

Please sign in to comment.