Format '.pyi' type stub files.

Closes #1151
google · Nov 6, 2023 · 486a5a2 · 486a5a2
1 parent d004965
commit 486a5a2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -6,6 +6,7 @@
 ### Changes
 - Remove dependency on importlib-metadata
 - Remove dependency on tomli when using >= py311
+- Format '.pyi' type sub files.
 ### Fixed
 - Fix SPLIT_ARGUMENTS_WHEN_COMMA_TERMINATED for one-item named argument lists
   by taking precedence over SPLIT_BEFORE_NAMED_ASSIGNS.

diff --git a/yapf/yapflib/file_resources.py b/yapf/yapflib/file_resources.py
@@ -252,7 +252,7 @@ def IsIgnored(path, exclude):
 
 def IsPythonFile(filename):
   """Return True if filename is a Python file."""
-  if os.path.splitext(filename)[1] == '.py':
+  if os.path.splitext(filename)[1] in frozenset({'.py', '.pyi'}):
     return True
 
   try: