Alter logic

niceplots/utils.py

@@ -260,7 +260,7 @@ def label_line_ends(ax, lines=None, labels=None, colors=None, x_offset_pts=6, y_
     ax : Matplotlib axes
         axes to label the lines of
     lines : iterable of matplotlib line objects, optional
-        Lines to label, by default all lines in the axes whose labels don't begin with an underscore
+        Lines to label, by default all lines in the axes
     labels : list of strings, optional
         Labels for each line, by default uses each line's label attribute
     colors : single or list of colors, optional
@@ -279,13 +279,17 @@ def label_line_ends(ax, lines=None, labels=None, colors=None, x_offset_pts=6, y_
     list of matplotlib annotation objects
         The annotations created
     """
+
     # By default label all lines in the plot whose labels don't begin with an underscore
     if lines is None:
-        lines = [line for line in ax.get_lines() if line.get_label()[0] != "_"]
+        lines = [line for line in ax.get_lines()]
 
     if labels is None:
+        # If we're not given labels, use the liness label attribute, but ignore any lines that are still using the
+        # matplotlib default label which starts with an underscore
+        lines = [line for line in lines if not line.get_label().startswith("_")]
         labels = [line.get_label() for line in lines]
-    
+
     numLines = len(lines)
     if len(labels) != numLines:
         raise ValueError(f"Number of labels ({len(labels)}) doesn't match number of lines ({numLines})")