Fix missing rounding from rebase

Author: jakubno

chart_data_extractor/e2b_charts/graphs/bars.py

@@ -4,6 +4,7 @@
 from pydantic import BaseModel, Field
 
 from .base import Graph2D, GraphType
+from ..utils.rounding import dynamic_round
 
 
 class BarData(BaseModel):
@@ -48,6 +49,7 @@ class BoxAndWhiskerData(BaseModel):
     median: float
     third_quartile: float
     max: float
+    outliers: List[float]
 
 
 class BoxAndWhiskerGraph(Graph2D):
@@ -56,22 +58,23 @@ class BoxAndWhiskerGraph(Graph2D):
     elements: List[BoxAndWhiskerData] = Field(default_factory=list)
 
     def _extract_info(self, ax: Axes) -> None:
-        super()._extract_info(ax)
+        labels = [item.get_text() for item in ax.get_xticklabels()]
 
         boxes = []
-        for box in ax.patches:
+        for label, box in zip(labels, ax.patches):
             vertices = box.get_path().vertices
-            x_vertices = vertices[:, 0]
-            y_vertices = vertices[:, 1]
+            x_vertices = [dynamic_round(x) for x in vertices[:, 0]]
+            y_vertices = [dynamic_round(y) for y in vertices[:, 1]]
             x = min(x_vertices)
             y = min(y_vertices)
             boxes.append(
                 {
                     "x": x,
                     "y": y,
-                    "label": box.get_label(),
-                    "width": round(max(x_vertices) - x, 4),
-                    "height": round(max(y_vertices) - y, 4),
+                    "label": label,
+                    "width": max(x_vertices) - x,
+                    "height": max(y_vertices) - y,
+                    "outliers": [],
                 }
             )
 
@@ -85,13 +88,21 @@ def _extract_info(self, ax: Axes) -> None:
                 box["x"], box["y"] = box["y"], box["x"]
                 box["width"], box["height"] = box["height"], box["width"]
 
-        for line in ax.lines:
-            xdata = line.get_xdata()
-            ydata = line.get_ydata()
+        for i, line in enumerate(ax.lines):
+            xdata = [dynamic_round(x) for x in line.get_xdata()]
+            ydata = [dynamic_round(y) for y in line.get_ydata()]
 
             if orientation == "vertical":
                 xdata, ydata = ydata, xdata
 
+            if len(xdata) == 1:
+                for box in boxes:
+                    if box["x"] <= xdata[0] <= box["x"] + box["width"]:
+                        break
+                else:
+                    continue
+
+                box["outliers"].append(ydata[0])
             if len(ydata) != 2:
                 continue
             for box in boxes:
@@ -101,6 +112,7 @@ def _extract_info(self, ax: Axes) -> None:
                 continue
 
             if (
+                # Check if the line is inside the box, prevent floating point errors
                 ydata[0] == ydata[1]
                 and box["y"] <= ydata[0] <= box["y"] + box["height"]
             ):
@@ -122,6 +134,7 @@ def _extract_info(self, ax: Axes) -> None:
                 median=box["median"],
                 third_quartile=box["y"] + box["height"],
                 max=box["whisker_upper"],
+                outliers=box["outliers"],
             )
             for box in boxes
         ]

chart_data_extractor/e2b_charts/graphs/pie.py

@@ -1,9 +1,11 @@
+from decimal import Decimal
 from typing import Literal, List
 
 from matplotlib.axes import Axes
 from pydantic import BaseModel, Field
 
 from .base import Graph, GraphType
+from ..utils.rounding import dynamic_round
 
 
 class PieData(BaseModel):
@@ -23,7 +25,7 @@ def _extract_info(self, ax: Axes) -> None:
         for wedge in ax.patches:
             pie_data = PieData(
                 label=wedge.get_label(),
-                angle=abs(round(wedge.theta2 - wedge.theta1, 4)),
+                angle=abs(dynamic_round(Decimal(wedge.theta2) - Decimal(wedge.theta1))),
                 radius=wedge.r,
             )
 

chart_data_extractor/e2b_charts/utils/rounding.py

@@ -0,0 +1,13 @@
+from decimal import Decimal, localcontext
+
+
+def dynamic_round(number):
+    # Convert to Decimal for precise control
+    decimal_number = Decimal(str(number))
+
+    # Dynamically determine precision based on magnitude
+    precision = max(1, 16 - decimal_number.adjusted())  # 16 digits of precision
+
+    with localcontext() as ctx:
+        ctx.prec = precision  # Set the dynamic precision
+        return +decimal_number  # The + operator applies rounding