From 59fd7704cfcbabf8ccb5f0cfaf2376a01551e74d Mon Sep 17 00:00:00 2001
From: Nikita Taranov <nikita.taranov@clickhouse.com>
Date: Wed, 14 Dec 2022 14:55:14 +0100
Subject: [PATCH] Support complex names

* input size is the last part of the name
* label is everything before the last slash
---
 plot.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/plot.py b/plot.py
index 6d09de8..54693fa 100644
--- a/plot.py
+++ b/plot.py
@@ -94,7 +94,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):
@@ -114,7 +114,7 @@ def read_data(args):
             'Could not parse the benchmark data. Did you forget "--benchmark_format=[csv|json] when running the benchmark"?'
         )
         exit(1)
-    data["label"] = data["name"].apply(lambda x: x.split("/")[0])
+    data["label"] = data["name"].apply(lambda x: x.rsplit("/", 1)[0])
     data["input"] = data["name"].apply(parse_input_size)
     data[args.metric] = data[args.metric].apply(TRANSFORMS[args.transform])
     return data