Set Instance Type on CLI in createami command

Signed-off-by: Sean Smith <seaam@amazon.com>
aws · Jun 27, 2019 · 732d628 · 732d628
1 parent a021ffd
commit 732d628
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/cli/pcluster/cli.py b/cli/pcluster/cli.py
@@ -313,6 +313,13 @@ def _get_parser():
         help="Specifies the OS of the base AMI. "
         "Valid options are: alinux, ubuntu1404, ubuntu1604, centos6, centos7.",
     )
+    pami.add_argument(
+        "-i",
+        "--instance-type",
+        dest="instance_type",
+        default="t2.xlarge",
+        help="Sets instance type to build the ami on. Defaults to t2.xlarge.",
+    )
     pami.add_argument(
         "-ap",
         "--ami-name-prefix",

diff --git a/cli/pcluster/pcluster.py b/cli/pcluster/pcluster.py
@@ -1033,7 +1033,7 @@ def create_ami(args):
     LOGGER.debug("Building AMI based on args %s", str(args))
     results = {}
 
-    instance_type = "t2.large"
+    instance_type = args.instance_type
     try:
         config = cfnconfig.ParallelClusterConfig(args)