Fix issue with type comments on lines with trailing commas (#1058)

[etc/vim.git] / blackd.py

diff --git a/blackd.py b/blackd.py
index 25a9a1589ac9102212fde79aed690a2afad2a153..37fcec31ec10b092a8a273dc03a5d59921d3464c 100644 (file)
--- a/blackd.py
+++ b/blackd.py
@@ -129,7 +129,11 @@ def parse_python_variant_header(value: str) -> Tuple[bool, Set[black.TargetVersi
         for version in value.split(","):
             if version.startswith("py"):
                 version = version[len("py") :]
         for version in value.split(","):
             if version.startswith("py"):
                 version = version[len("py") :]

-            major_str, *rest = version.split(".")
+            if "." in version:
+                major_str, *rest = version.split(".")
+            else:
+                major_str = version[0]
+                rest = [version[1:]] if len(version) > 1 else []

             try:
                 major = int(major_str)
                 if major not in (2, 3):
             try:
                 major = int(major_str)
                 if major not in (2, 3):