diff --git a/tools/perf/benchmarks b/tools/perf/benchmarks index f46b920463..df4c87b3b2 100755 --- a/tools/perf/benchmarks +++ b/tools/perf/benchmarks @@ -460,7 +460,7 @@ benchmarks: parser.add_argument("--benchmark", nargs="*", default=[b.id for b in self._benchmarks], metavar="BENCHMARKS", help="Benchmarks to run. Default suite will be run if omitted.") - parser.add_argument("--dist-one", type=str, + parser.add_argument("--dist-one", action="store_true", help="Copy logs and metrics to the given dist dir. Requires that only" + " one benchmark be supplied. Postroll steps will be skipped.") @@ -478,7 +478,7 @@ benchmarks: self._error(f"Invalid benchmark: {id}") # --dist-one requires that only one benchmark be supplied - if len(self.Benchmarks()) != 1: + if self._args.dist_one and len(self.Benchmarks()) != 1: self._error("--dist-one requires that exactly one --benchmark.") if self._had_error: