Fixed legend

EPC-MSU · Dec 14, 2023 · 0514eb7 · 0514eb7
1 parent 5295fb6
commit 0514eb7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/report_generator/plot.py b/report_generator/plot.py
@@ -165,6 +165,12 @@ def draw_ivc_for_pin(pin_info: PinInfo, index: int, file_name: str, scaling_type
     else:
         test_curve.clear_curve()
 
+    for curve in viewer.plot.curves:
+        if curve.curve is None:
+            curve.detach()
+        else:
+            curve.attach(viewer.plot)
+
     viewer.plot.grab().save(file_name, format="PNG")