update

bytedance · Aug 11, 2024 · bd08baf · bd08baf
1 parent a2dc250
commit bd08baf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/bench.py b/scripts/bench.py
@@ -169,7 +169,7 @@ def bench_diff(main, target, threshold=0.05):
 def main():
     argparser = argparse.ArgumentParser(description='Tools to test the performance. Example: ./bench.py "go test -bench=. ./..."')
     argparser.add_argument('cmd', type=str, help='Golang benchmark command')
-    argparser.add_argument('-nr', type=bool, dest='norestore', default=False, help='not restore branch')
+    argparser.add_argument('-nr', dest='norestore', action='store_true', help='not restore branch')
     argparser.add_argument('-t', type=float, dest='threshold', default=0.1, help='diff bench threshold')
     argparser.add_argument('-c', '--compare', dest='compare', action='store_true',
         help='Compare the current branch with the main branch')