From 94d760af6fd7c050f0ccc7390322f29349d2bbe5 Mon Sep 17 00:00:00 2001 From: Dvir Yitzchaki Date: Thu, 19 Sep 2019 21:22:34 -0600 Subject: [PATCH] support BENCHMARK_CAPTURE macro --- plot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/plot.py b/plot.py index 796c160..cab7c57 100644 --- a/plot.py +++ b/plot.py @@ -66,7 +66,7 @@ def parse_input_size(name): splits = name.split('/') if len(splits) == 1: return 1 - return int(splits[1]) + return int(splits[-1]) def read_data(args): @@ -77,7 +77,7 @@ def read_data(args): msg = 'Could not parse the benchmark data. Did you forget "--benchmark_format=csv"?' logging.error(msg) exit(1) - data['label'] = data['name'].apply(lambda x: x.split('/')[0]) + data['label'] = data['name'].apply(lambda x: x.split('/')[-2]) data['input'] = data['name'].apply(parse_input_size) data[args.metric] = data[args.metric].apply(TRANSFORMS[args.transform]) return data