Fix black compatibility for .pyi type stub files

PyCQA · Dec 6, 2022 · c098db3 · c098db3
1 parent 12cc5fb
commit c098db3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/isort/output.py b/isort/output.py
@@ -209,9 +209,10 @@ def sorted_imports(
                     break
 
             if config.lines_after_imports != -1:
-                formatted_output[imports_tail:0] = [
-                    "" for line in range(config.lines_after_imports)
-                ]
+                lines_after_imports = config.lines_after_imports
+                if config.profile == "black" and extension == "pyi":  # special case for black
+                    lines_after_imports = 1
+                formatted_output[imports_tail:0] = ["" for line in range(lines_after_imports)]
             elif extension != "pyi" and next_construct.startswith(STATEMENT_DECLARATIONS):
                 formatted_output[imports_tail:0] = ["", ""]
             else: