From 6b9aede77e2521f28c3f5839181a04626e3c3342 Mon Sep 17 00:00:00 2001
From: "Adam J. Stewart" <ajstewart426@gmail.com>
Date: Mon, 13 Sep 2021 15:34:41 +0000
Subject: [PATCH] Improve help message formatting

---
 benchmark.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/benchmark.py b/benchmark.py
index baf7b532a0b..a0f081e828c 100755
--- a/benchmark.py
+++ b/benchmark.py
@@ -55,6 +55,7 @@ def set_up_parser() -> argparse.ArgumentParser:
     group.add_argument(
         "-n",
         "--num-batches",
+        default=2 ** 7,
         type=int,
         help="number of batches to load",
         metavar="SIZE",
@@ -100,12 +101,13 @@ def set_up_parser() -> argparse.ArgumentParser:
         default="benchmark-results.csv",
         type=str,
         help="path to the CSV file to write results",
+        metavar="FILE",
     )
     parser.add_argument(
         "-v",
         "--verbose",
         action="store_true",
-        help="prints results to stdout",
+        help="print results to stdout",
     )
 
     return parser
@@ -184,7 +186,7 @@ def main(args: argparse.Namespace) -> None:
 
         tic = time.time()
         for i, batch in enumerate(dataloader):
-            # this is to stop the GridGeoSampler from enumerating everything
+            # This is to stop the GridGeoSampler from enumerating everything
             if i == num_batches:
                 break
         toc = time.time()