Merge pull request #13074 from wilsonwang371/rw-bm-bugfix

tools: fix small rw-heatmaps python plot script argument parsing bug
etcd-io · Jun 2, 2021 · af6f3dc · af6f3dc
2 parents bd475d8 + d3ae6f2
commit af6f3dc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/rw-heatmaps/plot_data.py b/tools/rw-heatmaps/plot_data.py
@@ -23,9 +23,11 @@ def parse_args():
                         help='second input data files in csv format. (optional)')
     parser.add_argument('-t', '--title', dest='title', type=str, required=True,
                         help='plot graph title string')
-    parser.add_argument('-z', '--zero-centered', dest='zero', type=bool, required=False,
+    parser.add_argument('-z', '--zero-centered', dest='zero', action='store_true', required=False,
                         help='plot the improvement graph with white color represents 0.0',
                         default=True)
+    parser.add_argument('--no-zero-centered', dest='zero', action='store_false', required=False,
+                        help='plot the improvement graph without white color represents 0.0')
     parser.add_argument('-o', '--output-image', dest='output', type=str, required=True,
                         help='output image filename')
     return parser.parse_args()