fix deprecated tostring_rgb

aimclub · Dec 23, 2024 · 841ad41 · 841ad41
1 parent 7595be9
commit 841ad41
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/golem/visualisation/opt_viz_extra.py b/golem/visualisation/opt_viz_extra.py
@@ -279,8 +279,8 @@ def extract_objectives(individuals: List[List[Any]], objectives_numbers: Tuple[i
 
 
 def figure_to_array(fig):
-    img = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8)
-    img = img.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+    img = np.frombuffer(fig.canvas.buffer_rgba(), dtype=np.uint8)
+    img = img.reshape(fig.canvas.get_width_height()[::-1] + (4,))
     return img