Don't leave invalid trailing comma on imports

[etc/vim.git] / black.py

diff --git a/black.py b/black.py
index 46754566fd4aaac239b93c0250fd9c12da2c9d39..fd2b75ed2c1cf9d5ffb3759f142b6855b0a23b9e 100644 (file)
--- a/black.py
+++ b/black.py
@@ -885,9 +885,14 @@ class Line:
                 self.remove_trailing_comma()
                 return True
 
                 self.remove_trailing_comma()
                 return True
 

-        # For parens let's check if it's safe to remove the comma.  If the
-        # trailing one is the only one, we might mistakenly change a tuple
-        # into a different type by removing the comma.
+        # For parens let's check if it's safe to remove the comma.
+        # Imports are always safe.
+        if self.is_import:
+            self.remove_trailing_comma()
+            return True
+
+        # Otheriwsse, if the trailing one is the only one, we might mistakenly
+        # change a tuple into a different type by removing the comma.

         depth = closing.bracket_depth + 1
         commas = 0
         opening = closing.opening_bracket
         depth = closing.bracket_depth + 1
         commas = 0
         opening = closing.opening_bracket