From c098db304f3bd9c6eb098be63f74ab25fe1a7b78 Mon Sep 17 00:00:00 2001 From: Xuehai Pan Date: Wed, 7 Dec 2022 02:47:55 +0800 Subject: [PATCH] Fix `black` compatibility for `.pyi` type stub files --- isort/output.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/isort/output.py b/isort/output.py index c59be936d..59143431e 100644 --- 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: