Fix labels

e2b-dev · Sep 30, 2024 · 32af456 · 32af456
1 parent 77c67ce
commit 32af456
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/template/startup_scripts/0002_data.py b/template/startup_scripts/0002_data.py
@@ -119,7 +119,7 @@ def _extract_info(self, ax: Axes) -> None:
         for line in ax.get_lines():
             label = line.get_label()
             if label.startswith("_child"):
-                number = int(label[7:])
+                number = int(label[6:])
                 label = f"Line {number}"
 
             points = [(x, y) for x, y in zip(line.get_xdata(), line.get_ydata())]
@@ -155,7 +155,7 @@ def _extract_info(self, ax: Axes) -> None:
         for container in ax.containers:
             group_label = container.get_label()
             if group_label.startswith("_container"):
-                number = int(group_label[11:])
+                number = int(group_label[10:])
                 group_label = f"Group {number}"
 
             heights = [rect.get_height() for rect in container]