X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/ad1696422b7343a0cfea3c4b304e6acfa15d3cff..a18c7bc09942951e93cbd142fb7384aa2af18951:/src/black/__init__.py

diff --git a/src/black/__init__.py b/src/black/__init__.py
index 1c69cc4..49d088b 100644
--- a/src/black/__init__.py
+++ b/src/black/__init__.py
@@ -755,6 +755,8 @@ def reformat_one(
             is_stdin = False
 
         if is_stdin:
+            if src.suffix == ".pyi":
+                mode = replace(mode, is_pyi=True)
             if format_stdin_to_stdout(fast=fast, write_back=write_back, mode=mode):
                 changed = Changed.YES
         else: