Matplotlib export: avoid numpy repr because of version differences

Author: markotoplak

Orange/widgets/utils/matplotlib_export.py

@@ -11,13 +11,23 @@
 def numpy_repr(a):
     """ A numpy repr without summarization """
     opts = np.get_printoptions()
+    # avoid numpy repr as it changes between versions
+    # TODO handle numpy repr differences
+    if isinstance(a, np.ndarray):
+        return "array(" + repr(list(a)) + ")"
     try:
         np.set_printoptions(threshold=10**10)
         return repr(a)
     finally:
         np.set_printoptions(**opts)
 
 
+def numpy_repr_int(a):
+    # avoid numpy repr as it changes between versions
+    # TODO handle numpy repr differences
+    return "array(" + repr(list(a)) + ", dtype='int')"
+
+
 def compress_if_all_same(l):
     s = set(l)
     return s.pop() if len(s) == 1 else l
@@ -113,7 +123,7 @@ def do_colors(code, data_column, show, name):
 
         code.append("{} = {}".format(name, repr(colors)))
         if index is not None:
-            code.append("{}_index = {}".format(name, repr(index)))
+            code.append("{}_index = {}".format(name, numpy_repr_int(index)))
 
         decompresssed_code = name
         if index is not None:
@@ -153,7 +163,7 @@ def matplotlib_marker(m):
         if np.all(indices == np.arange(x.shape[0])):
             indices = None
         if indices is not None:
-            code.append("indices = {}".format(numpy_repr(indices)))
+            code.append("indices = {}".format(numpy_repr_int(indices)))
 
         def indexed(data, data_name, indices=indices):
             return code_with_indices(data, data_name, indices, "indices")