Basic changes

CharlesHe16 · CharlesHe16 · commit 68f91d560ec7 · 2020-12-27T22:38:07.000-08:00
diff --git a/lib/matplotlib/sankey.py b/lib/matplotlib/sankey.py
@@ -739,7 +739,14 @@ def _get_angle(a, r):
             if label is None or angle is None:
                 label = ''
             elif self.unit is not None:
-                quantity = self.format % abs(number) + self.unit
+                if self.format is None:
+                    self.format = '%G'
+                if isinstance(self.format, str):
+                    quantity = self.format % abs(number) + self.unit
+                elif callable(self.format):
+                    quantity = self.format(100. * abs(number))
+                else:
+                    raise TypeError('format must be callable or a format string')
                 if label != '':
                     label += "\n"
                 label += quantity
