Skip to content

Commit

Permalink
Improve help message formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
adamjstewart committed Sep 16, 2021
1 parent ec38ea3 commit 6b9aede
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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()
Expand Down

0 comments on commit 6b9aede

Please sign in to comment.