From ad4bef711d159b922696d510d999a25bda50c5d4 Mon Sep 17 00:00:00 2001
From: Xi Chen <putcncx@gmail.com>
Date: Tue, 3 Apr 2018 22:20:05 -0700
Subject: [PATCH] move required arguments together

---
 .../paddle_banchmarking_aws.py                | 28 ++++++++++---------
 1 file changed, 15 insertions(+), 13 deletions(-)

diff --git a/tools/aws_benchmarking/paddle_banchmarking_aws.py b/tools/aws_benchmarking/paddle_banchmarking_aws.py
index f6512d923ce..c63e4b0742e 100644
--- a/tools/aws_benchmarking/paddle_banchmarking_aws.py
+++ b/tools/aws_benchmarking/paddle_banchmarking_aws.py
@@ -28,6 +28,19 @@ import paramiko
 # ~/.aws/credentials and ~/.aws/config
 
 parser = argparse.ArgumentParser(description=__doc__)
+parser.add_argument(
+    '--key_name',
+    type=str,
+    default="",
+    required=True,
+    help="required, key pair name")
+parser.add_argument(
+    '--security_group_id',
+    type=str,
+    default="",
+    required=True,
+    help="required, the security group id associated with your VPC")
+
 parser.add_argument(
     '--vpc_id',
     type=str,
@@ -38,12 +51,7 @@ parser.add_argument(
     type=str,
     default="",
     help="The Subnet_id in which you wish to run test")
-parser.add_argument(
-    '--security_group_id',
-    type=str,
-    default="",
-    required=True,
-    help="required, the security group id associated with your VPC")
+
 parser.add_argument(
     '--pserver_instance_type',
     type=str,
@@ -54,12 +62,7 @@ parser.add_argument(
     type=str,
     default="p2.xlarge",
     help="your trainer instance type")
-parser.add_argument(
-    '--key_name',
-    type=str,
-    default="",
-    required=True,
-    help="required, key pair name")
+
 parser.add_argument(
     '--task_name',
     type=str,
@@ -316,7 +319,6 @@ def cleanup(task_name):
     }])
 
     ec2client.delete_subnet(SubnetId=subnet["Subnets"][0]["SubnetId"])
-
     # no subnet delete waiter, just leave it.
     return
 
-- 
GitLab