Add __format__ method to functional units

astropy · Feb 16, 2023 · 0f76ee8 · 0f76ee8
1 parent 48bc134
commit 0f76ee8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/astropy/units/function/core.py b/astropy/units/function/core.py
@@ -430,6 +430,13 @@ def to_string(self, format="generic"):
                 self_str = "\n".join(lines)
         return self_str
 
+    def __format__(self, format_spec):
+        """Try to format units using a formatter."""
+        try:
+            return self.to_string(format=format_spec)
+        except ValueError:
+            return format(str(self), format_spec)
+
     def __str__(self):
         """Return string representation for unit."""
         self_str = str(self.function_unit)