diff --git a/CHANGELOG.md b/CHANGELOG.md index 0063a54..d09db00 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -5,6 +5,7 @@ - Renamed CLI `--ci` flag to `--check`. - Removed Unicode interpuct from CLI output. - Removed colors from CLI output when CI envvar is set. +- Added support for `.pyi` files. ## v1.1.1 diff --git a/src/unbeheader/__init__.py b/src/unbeheader/__init__.py index c7b81b0..95c5dfa 100644 --- a/src/unbeheader/__init__.py +++ b/src/unbeheader/__init__.py @@ -10,6 +10,9 @@ 'py': SupportedFileType( re.compile(r'((^#|[\r\n]#).*)*'), CommentSkeleton('#', '#')), + 'pyi': SupportedFileType( + re.compile(r'((^#|[\r\n]#).*)*'), + CommentSkeleton('#', '#')), 'wsgi': SupportedFileType( re.compile(r'((^#|[\r\n]#).*)*'), CommentSkeleton('#', '#')),