fix to pass ci lint (flake8)

(see #404 (comment))

Author: senselessDev

test/test_cleanfigure.py

@@ -1,7 +1,7 @@
 import numpy as np
-import pytest
 from matplotlib import pyplot as plt
 
+import pytest
 from tikzplotlib import clean_figure, get_tikz_code
 
 RC_PARAMS = {"figure.figsize": [5, 5], "figure.dpi": 220, "pgf.rcfonts": False}

test/test_rotated_labels.py

@@ -1,9 +1,9 @@
 import os
 import tempfile
 
-import pytest
 from matplotlib import pyplot as plt
 
+import pytest
 import tikzplotlib
 
 

tikzplotlib/_util.py

@@ -13,8 +13,8 @@ def get_legend_text(obj):
     if leg is None:
         return None
 
-    keys = [l.get_label() for l in leg.legendHandles if l is not None]
-    values = [l.get_text() for l in leg.texts]
+    keys = [h.get_label() for h in leg.legendHandles if h is not None]
+    values = [t.get_text() for t in leg.texts]
 
     label = obj.get_label()
     d = dict(zip(keys, values))