ratio1 3.4.125__tar.gz → 3.4.126__tar.gz

{ratio1-3.4.125 → ratio1-3.4.126}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ratio1
-Version: 3.4.125
+Version: 3.4.126
 Summary: `ratio1` or Ration1 SDK is the Python SDK required for client app development for the Ratio1 ecosystem
 Project-URL: Homepage, https://github.com/Ratio1/ratio1_sdk
 Project-URL: Bug Tracker, https://github.com/Ratio1/ratio1_sdk/issues

{ratio1-3.4.125 → ratio1-3.4.126}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "ratio1"
-version = "3.4.125"
+version = "3.4.126"
 authors = [
   { name="Andrei Ionut Damian", email="andrei.damian@ratio1.ai" },
   { name="Cristan Bleotiu", email="cristian.bleotiu@ratio1.ai" },

{ratio1-3.4.125 → ratio1-3.4.126}/ratio1/_ver.py

@@ -1,4 +1,4 @@
-__VER__ = "3.4.125"
+__VER__ = "3.4.126"
 
 if __name__ == "__main__":
   with open("pyproject.toml", "rt") as fd:

{ratio1-3.4.125 → ratio1-3.4.126}/ratio1/logging/logger_mixins/computer_vision_mixin.py

@@ -270,10 +270,18 @@ class _ComputerVisionMixin(object):
       plt.axis('off')
     fig = plt.gcf()
     fig.canvas.draw()
-    data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
-    w, h = fig.canvas.get_width_height()
     try:
-      np_img = data.reshape((int(h), int(w), -1))
+      w, h = fig.canvas.get_width_height()
+      if hasattr(fig.canvas, 'tostring_rgb'):
+        data = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+        np_img = data.reshape((int(h), int(w), 3))
+      elif hasattr(fig.canvas, 'tostring_argb'):
+        data = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8)
+        argb = data.reshape((int(h), int(w), 4))
+        np_img = argb[:, :, 1:4]
+      else:
+        rgba = np.asarray(fig.canvas.buffer_rgba())
+        np_img = rgba[:, :, :3]
     except Exception as e:
       print(e)
       np_img = None
@@ -440,4 +448,4 @@ class _ComputerVisionMixin(object):
     resized = cv2.resize(image, dim, interpolation = inter)
 
     # return the resized image
-    return resized
+    return resized