diff --git a/main.py b/main.py index 7ea89d3ad9d7a6dc5bab0f4df49c33c2e0a2dce2..42a4b7cf4579f32c6937259d24198e64bc20d5df 100755 --- a/main.py +++ b/main.py @@ -19,27 +19,28 @@ def get_argument_parser(): ) parser.add_argument( - "--kernel", "-k", type=str, required=False, help="Specify the kernel name" + "-k", "--kernel", type=str, required=False, help="Specify the kernel name" ) parser.add_argument( - "--mock", "-m", + "--mock", type=str, help="Specify mock input file to use instead of profiling output", ) parser.add_argument( - "--output", "-o", type=str, help="Write full profiling output to file" + "-o", "--output", type=str, help="Write full profiling output to file" ) parser.add_argument( - "--quiet", "-q", + "--quiet", default=False, action="store_true", help="Print profiling output only", ) + return parser