diff --git a/windrose/windrose.py b/windrose/windrose.py index 65d5ffd..83f4763 100644 --- a/windrose/windrose.py +++ b/windrose/windrose.py @@ -264,7 +264,10 @@ def get_labels(decimal_places=1, units=None): labels = [ fmt.format(digits[k], digits[k + 1]) for k in range(len(digits) - 1) ] - labels[-1] = f">{digits[-1]}" + if units: + labels[-1] = f">{digits[-1]} " + units + else: + labels[-1] = f">{digits[-1]}" return labels kwargs.pop("labels", None)