legend colors initialization moved to save.py

Author: danielhkl

matplotlib2tikz/legend.py

@@ -200,15 +200,14 @@ def draw_legend(data, obj):
             )
 
     # Set color of lines in legend
-    data['legend colors'] = []
     for handle in obj.legendHandles:
         try:
             data, legend_color, _ = mycol.mpl_color2xcolor(data,
                                                             handle.get_color())
             data['legend colors'].append('\\addlegendimage{no markers, %s}\n'
                                         % legend_color)
         except AttributeError:
-            data['legend colors'] = []
+            pass
 
     # Write styles to data
     if legend_style:

matplotlib2tikz/save.py

@@ -124,6 +124,7 @@ def get_tikz_code(
     data['pgfplots libs'] = set()
     data['font size'] = textsize
     data['custom colors'] = {}
+    data['legend colors'] = []
     data['extra tikzpicture parameters'] = extra_tikzpicture_parameters
     # rectangle_legends is used to keep track of which rectangles have already
     # had \addlegendimage added. There should be only one \addlegenimage per