correction of 'seperated parameter extra in extra_axis_opt and extra_picture_opt'

Author: danielhkl

matplotlib2tikz/save.py

@@ -26,8 +26,8 @@ def get_tikz_code(
         tex_relative_path_to_data=None,
         strict=False,
         wrap=True,
-        extra_axis_opt=None,
-        extra_picture_opt=None,
+        extra_axis_parameters=set(),
+        extra_tikzpicture_parameters=set(),
         dpi=None,
         show_info=True
         ):
@@ -84,13 +84,13 @@ def get_tikz_code(
                  Default is ``True``.
     :type wrap: bool
 
-    :param extra_axis_opt: Extra axis options to be passed (as a set) to pgfplots.
+    :param extra_axis_parameters: Extra axis options to be passed (as a set) to pgfplots.
                   Default is ``None``.
-    :type extra_axis_opt: a set of strings for the pfgplots axes.
+    :type extra_axis_parameters: a set of strings for the pfgplots axes.
 
-    :param extra_picture_opt: Extra tikzpicture options to be passed (as a set) to pgfplots.
-                  Default is ``None``.
-    :type extra_picture_opt: a set of strings for the pfgplots tikzpicture.
+    :param extra_tikzpicture_parameters: Extra tikzpicture options to be passed (as a set) to pgfplots.
+
+    :type extra_tikzpicture_parameters: a set of strings for the pfgplots tikzpicture.
 
     :param dpi: The resolution in dots per inch of the rendered image in case
                 of QuadMesh plots. If ``None`` it will default to the value
@@ -122,19 +122,15 @@ def get_tikz_code(
     data['pgfplots libs'] = set()
     data['font size'] = textsize
     data['custom 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
     # bar chart data series.
     data['rectangle_legends'] = set()
-    if extra_axis_opt:
-        data['extra axis options [base]'] = extra_axis_opt.copy()
-    else:
-        data['extra axis options [base]'] = set()
-
-    if extra_picture_opt:
-        data['extra picture options'] = extra_picture_opt.copy()
-    else:
-        data['extra picture options'] = set()
+    #if extra_axis_parameters:
+    data['extra axis options [base]'] = extra_axis_parameters.copy()
+    #else:
+     #   data['extra axis options [base]'] = set()
 
     if dpi is None:
         savefig_dpi = mpl.rcParams['savefig.dpi']
@@ -162,10 +158,8 @@ def get_tikz_code(
     # write the contents
     if wrap:
         code += '\\begin{tikzpicture}\n\n'
-        if extra_picture_opt:
-            for options in data['extra picture options']:
-                code += options + ',\n'
-
+        code += ',\n'.join(data['extra tikzpicture parameters'])
+        code += '\n'
 
     coldefs = _get_color_definitions(data)
     if coldefs: